Support torch 1.10.x (#914)

* Support torch 1.10.x

* Fix installing PyTorch.

Author: csukuangfj

.github/workflows/build-cpu.yml

@@ -36,8 +36,8 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-18.04, macos-10.15]
-        torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10"]
-        # Python 3.9 is for PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10
+        torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
+        # Python 3.9 is for PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10.x
         python-version: [3.6, 3.7, 3.8, 3.9]
         exclude:
           - python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]

.github/workflows/build.yml

@@ -39,7 +39,7 @@ jobs:
         # from https://download.pytorch.org/whl/torch_stable.html
         # Note: There are no torch versions for CUDA 11.2
         #
-        # 1.10 supports: cuda10.2 (default), 11.1, 11.3
+        # 1.10.x supports: cuda10.2 (default), 11.1, 11.3
         # 1.9.x supports: cuda10.2 (default), 11.1
         # PyTorch 1.8.x supports: cuda 10.1, 10.2 (default), 11.1
         # PyTorch 1.7.x supports: cuda 10.1, 10.2 (default), 11.0
@@ -49,9 +49,9 @@ jobs:
         # CUDA 11.3 is for torch 1.10
         cuda: ["10.1", "10.2", "11.0", "11.1", "11.3"]
         gcc: ["7"]
-        torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10"]
+        torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
         #
-        # Python 3.9 is for PyTorch 1.7.1, 1.8.0, 1.8.1, 1.9.x, 1.10
+        # Python 3.9 is for PyTorch 1.7.1, 1.8.0, 1.8.1, 1.9.x, 1.10.x
         python-version: [3.6, 3.7, 3.8, 3.9]
         exclude:
           - cuda: "11.3" # exclude 11.3 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1]
@@ -72,7 +72,7 @@ jobs:
             torch: "1.9.0"
           - cuda: "11.3"
             torch: "1.9.1"
-          - cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10]
+          - cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
             torch: "1.5.0"
           - cuda: "11.0"
             torch: "1.5.1"
@@ -87,7 +87,11 @@ jobs:
           - cuda: "11.0"
             torch: "1.9.1"
           - cuda: "11.0"
-            torch: "1.10"
+            torch: "1.10.0"
+          - cuda: "11.0"
+            torch: "1.10.1"
+          - cuda: "11.0"
+            torch: "1.10.2"
           - cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1]
             torch: "1.5.0"
           - cuda: "11.1"
@@ -98,12 +102,16 @@ jobs:
             torch: "1.7.0"
           - cuda: "11.1"
             torch: "1.7.1"
-          - cuda: "10.1" # exclude CUDA 10.1 for [1.9.0, 1.9.1, 1.10]
+          - cuda: "10.1" # exclude CUDA 10.1 for [1.9.0, 1.9.1, 1.10.0, 10.1, 10.2]
             torch: "1.9.0"
           - cuda: "10.1"
             torch: "1.9.1"
           - cuda: "10.1"
-            torch: "1.10"
+            torch: "1.10.0"
+          - cuda: "10.1"
+            torch: "1.10.1"
+          - cuda: "10.1"
+            torch: "1.10.2"
           - python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
             torch: "1.5.0"
           - python-version: 3.9

.github/workflows/build_conda.yml

@@ -37,7 +37,7 @@ jobs:
         cuda: ["10.1", "10.2", "11.0", "11.1", "11.3"]
         # from https://download.pytorch.org/whl/torch_stable.html
         #
-        # PyTorch 1.10 supports: 10.2 (default), 11.1, 11.3
+        # PyTorch 1.10.x supports: 10.2 (default), 11.1, 11.3
         # PyTorch 1.9.x supports: 10.2 (default), 11.1
         # PyTorch 1.8.1 supports: cuda 10.1, 10.2 (default), 11.1
         # PyTorch 1.8.0 supports: cuda 10.1, 10.2 (default), 11.1
@@ -56,9 +56,9 @@ jobs:
         #  https://github.com/csukuangfj/k2/runs/2533830771?check_suite_focus=true
         #  and
         #  https://github.com/NVIDIA/apex/issues/805
-        torch: ["1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10"]
+        torch: ["1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
         exclude:
-          # - cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10]
+          # - cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
           #   torch: "1.5.0"
           # - cuda: "11.0"
           #   torch: "1.5.1"
@@ -73,7 +73,11 @@ jobs:
           - cuda: "11.0"
             torch: "1.9.1"
           - cuda: "11.0"
-            torch: "1.10"
+            torch: "1.10.0"
+          - cuda: "11.0"
+            torch: "1.10.1"
+          - cuda: "11.0"
+            torch: "1.10.2"
           # - cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1]
           #   torch: "1.5.0"
           # - cuda: "11.1"
@@ -84,12 +88,16 @@ jobs:
             torch: "1.7.0"
           - cuda: "11.1"
             torch: "1.7.1"
-          - cuda: "10.1" # exclude 10.1 for [1.9.0, 1.9.1, 1.10]
+          - cuda: "10.1" # exclude 10.1 for [1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
             torch: "1.9.0"
           - cuda: "10.1"
             torch: "1.9.1"
           - cuda: "10.1"
-            torch: "1.10"
+            torch: "1.10.0"
+          - cuda: "10.1"
+            torch: "1.10.1"
+          - cuda: "10.1"
+            torch: "1.10.2"
           - python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
             torch: "1.5.0"
           - python-version: 3.9

.github/workflows/build_conda_cpu.yml

@@ -51,7 +51,7 @@ jobs:
         #
         # Other PyTorch versions are not tested
         #
-        torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10"]
+        torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
         exclude:
           - python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
             torch: "1.5.0"

.github/workflows/nightly-cpu.yml

@@ -41,7 +41,7 @@ jobs:
         os: [ubuntu-18.04, macos-10.15]
         # Python 3.9 is for PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10
         python-version: [3.6, 3.7, 3.8, 3.9]
-        torch: ["1.4.0", "1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10"]
+        torch: ["1.4.0", "1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
         exclude:
           - python-version: 3.9 # exclude Python 3.9 for [1.4.0, 1.5.0, 1.5.1, 1.6.0, 1.7.0]
             torch: "1.4.0"

scripts/github_actions/install_torch.sh

@@ -91,7 +91,7 @@ case ${torch} in
         ;;
     esac
     ;;
-  1.10)
+  1.10.*)
     case ${cuda} in
       10.2)
         package="torch==${torch}"