Update for CI (pytorch#2741)

* Update for CI * Update pytorch-version-tests.yml * Update pytorch-version-tests.yml * Update pytorch-version-tests.yml
henrytseng · Oct 15, 2022 · 32ef11d · 32ef11d
1 parent 8d5d7ca
commit 32ef11d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/.github/workflows/gpu-tests.yml b/.github/workflows/gpu-tests.yml
@@ -56,7 +56,7 @@ jobs:
       - name: Install PyTorch
         # https://pytorch.org/get-started/locally/
         run: |
-          pip install torch torchvision -f https://download.pytorch.org/whl/cu113/torch_stable.html
+          pip install torch torchvision -f https://download.pytorch.org/whl/cu116/torch_stable.html
           nvidia-smi
           python -c "import torch; print('CUDA is available: ', torch.cuda.is_available())"
 

diff --git a/.github/workflows/pytorch-version-tests.yml b/.github/workflows/pytorch-version-tests.yml
@@ -14,9 +14,9 @@ jobs:
       max-parallel: 10
       fail-fast: false
       matrix:
-        python-version: [3.7, 3.8, 3.9]
+        python-version: [3.7, 3.8, 3.9, "3.10"]
         pytorch-version:
-          [1.11.0, 1.10.0, 1.9.1, 1.8.1, 1.7.1, 1.6.0, 1.5.1, 1.4.0, 1.3.1]
+          [1.12.1, 1.11.0, 1.10.0, 1.9.1, 1.8.1, 1.7.1, 1.6.0, 1.5.1, 1.4.0, 1.3.1]
         exclude:
           - pytorch-version: 1.3.1
             python-version: 3.8
@@ -34,6 +34,24 @@ jobs:
             python-version: 3.9
           - pytorch-version: 1.8.1
             python-version: 3.9
+          - pytorch-version: 1.3.1
+            python-version: 3.10
+          - pytorch-version: 1.4.0
+            python-version: 3.10
+          - pytorch-version: 1.5.1
+            python-version: 3.10
+          - pytorch-version: 1.6.0
+            python-version: 3.10
+          - pytorch-version: 1.7.1
+            python-version: 3.10
+          - pytorch-version: 1.8.1
+            python-version: 3.10
+          - pytorch-version: 1.9.1
+            python-version: 3.10
+          - pytorch-version: 1.10.0
+            python-version: 3.10
+          - pytorch-version: 1.11.0
+            python-version: 3.10
 
     steps:
       - uses: actions/checkout@v2