Various version combinations of CUDA/Python/Torch.

k2-fsa · Dec 8, 2021 · 75b82b9 · 75b82b9
1 parent d80fefc
commit 75b82b9
Show file tree

Hide file tree

Showing 3 changed files with 66 additions and 73 deletions.
diff --git a/.github/workflows/build_conda.yml b/.github/workflows/build_conda.yml
@@ -38,10 +38,8 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: [3.6]
-        # python-version: [3.6, 3.7, 3.8, 3.9]
-        cuda: ["10.1"]
-        # cuda: ["10.1", "10.2", "11.0", "11.1", "11.3"]
+        python-version: [3.6, 3.7, 3.8, 3.9]
+        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
@@ -63,49 +61,48 @@ jobs:
         #  https://github.com/csukuangfj/k2/runs/2533830771?check_suite_focus=true
         #  and
         #  https://github.com/NVIDIA/apex/issues/805
-        torch: ["1.7.1"]
-        # torch: ["1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10"]
-        # 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]
-        #   #   torch: "1.5.0"
-        #   # - cuda: "11.0"
-        #   #   torch: "1.5.1"
-        #   - cuda: "11.0"
-        #     torch: "1.6.0"
-        #   - cuda: "11.0"
-        #     torch: "1.8.0"
-        #   - cuda: "11.0"
-        #     torch: "1.8.1"
-        #   - cuda: "11.0"
-        #     torch: "1.9.0"
-        #   - cuda: "11.0"
-        #     torch: "1.9.1"
-        #   - cuda: "11.0"
-        #     torch: "1.10"
-        #   # - 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"
-        #   #   torch: "1.5.1"
-        #   - cuda: "11.1"
-        #     torch: "1.6.0"
-        #   - cuda: "11.1"
-        #     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]
-        #     torch: "1.9.0"
-        #   - cuda: "10.1"
-        #     torch: "1.9.1"
-        #   - cuda: "10.1"
-        #     torch: "1.10"
-        #   - 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
-        #     torch: "1.5.1"
-        #   - python-version: 3.9
-        #     torch: "1.6.0"
-        #   - python-version: 3.9
-        #     torch: "1.7.0"
+        torch: ["1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10"]
+        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]
+          #   torch: "1.5.0"
+          # - cuda: "11.0"
+          #   torch: "1.5.1"
+          - cuda: "11.0"
+            torch: "1.6.0"
+          - cuda: "11.0"
+            torch: "1.8.0"
+          - cuda: "11.0"
+            torch: "1.8.1"
+          - cuda: "11.0"
+            torch: "1.9.0"
+          - cuda: "11.0"
+            torch: "1.9.1"
+          - cuda: "11.0"
+            torch: "1.10"
+          # - 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"
+          #   torch: "1.5.1"
+          - cuda: "11.1"
+            torch: "1.6.0"
+          - cuda: "11.1"
+            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]
+            torch: "1.9.0"
+          - cuda: "10.1"
+            torch: "1.9.1"
+          - cuda: "10.1"
+            torch: "1.10"
+          - 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
+            torch: "1.5.1"
+          - python-version: 3.9
+            torch: "1.6.0"
+          - python-version: 3.9
+            torch: "1.7.0"
 
     steps:
       # refer to https://github.com/actions/checkout

diff --git a/.github/workflows/build_conda_cpu_macos.yml b/.github/workflows/build_conda_cpu_macos.yml
@@ -46,26 +46,24 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        # python-version: [3.6, 3.7, 3.8, 3.9]
-        python-version: [3.6]
+        python-version: [3.6, 3.7, 3.8, 3.9]
         # from https://download.pytorch.org/whl/torch_stable.html
         #
         # PyTorch 1.10, 1.9.x, 1.8.x, and 1.7.1 support 3.6, 3.7, 3.8, 3.9
         # PyTorch 1.7.0, 1.6.0, and 1.5.x support 3.6, 3.7, 3.8
         #
         # Other PyTorch versions are not tested
         #
-        torch: ["1.5.0"]
-        # 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"]
-        # 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"
-        #   - python-version: 3.9
-        #     torch: "1.5.1"
-        #   - python-version: 3.9
-        #     torch: "1.6.0"
-        #   - python-version: 3.9
-        #     torch: "1.7.0"
+        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"]
+        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"
+          - python-version: 3.9
+            torch: "1.5.1"
+          - python-version: 3.9
+            torch: "1.6.0"
+          - python-version: 3.9
+            torch: "1.7.0"
 
     steps:
       # refer to https://github.com/actions/checkout

diff --git a/.github/workflows/build_conda_cpu_ubuntu.yml b/.github/workflows/build_conda_cpu_ubuntu.yml
@@ -48,26 +48,24 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: [3.6]
-        # python-version: [3.6, 3.7, 3.8, 3.9]
+        python-version: [3.6, 3.7, 3.8, 3.9]
         # from https://download.pytorch.org/whl/torch_stable.html
         #
         # PyTorch 1.10, 1.9.x, 1.8.x, and 1.7.1 support 3.6, 3.7, 3.8, 3.9
         # PyTorch 1.7.0, 1.6.0, and 1.5.x support 3.6, 3.7, 3.8
         #
         # Other PyTorch versions are not tested
         #
-        torch: ["1.5.0"]
-        # 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"]
-        # 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"
-        #   - python-version: 3.9
-        #     torch: "1.5.1"
-        #   - python-version: 3.9
-        #     torch: "1.6.0"
-        #   - python-version: 3.9
-        #     torch: "1.7.0"
+        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"]
+        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"
+          - python-version: 3.9
+            torch: "1.5.1"
+          - python-version: 3.9
+            torch: "1.6.0"
+          - python-version: 3.9
+            torch: "1.7.0"
 
     steps:
       # refer to https://github.com/actions/checkout