specify torchvision ver

gau-nernst · Jul 7, 2023 · db28b4b · db28b4b
1 parent 7029ada
commit db28b4b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
@@ -13,6 +13,11 @@ jobs:
       matrix:
         pytorch-version: ["1.13", "2.0"]
         python-version: ["3.10"]
+        include:
+          - pytorch-version: "1.13"
+            torchvision-version: "0.14"
+          - pytorch-version: "2.0"
+            torchvision-version: "0.15"
 
     steps:
       - name: Checkout
@@ -25,7 +30,7 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip install torch~=${{ matrix.pytorch-version }} torchvision --extra-index-url https://download.pytorch.org/whl/cpu
+          pip install torch~=${{ matrix.pytorch-version }} torchvision~=${{ matrix.torchvision-version }} --extra-index-url https://download.pytorch.org/whl/cpu
           pip install pytest
 
       - name: Run tests