Update workflows for PyTorch 2.x

Signed-off-by: Beat Buesser <beat.buesser@ibm.com>
Trusted-AI · Dec 4, 2023 · 30de53c · 30de53c
1 parent e9fd86c
commit 30de53c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci-pytorch.yml b/.github/workflows/ci-pytorch.yml
@@ -38,7 +38,7 @@ jobs:
             framework: pytorch
             python: '3.10'
             torch: 2.1.1
-            torchvision: 2.1.1
+            torchvision: 0.16.1+cpu
             torchaudio: 2.1.1
 
     name: ${{ matrix.name }}

diff --git a/requirements_test.txt b/requirements_test.txt
@@ -33,7 +33,7 @@ mxnet-native==1.8.0.post0
 --find-links https://download.pytorch.org/whl/cpu/torch_stable.html
 torch==2.1.1
 torchaudio==2.1.1
-torchvision==2.1.1
+torchvision==0.16.1+cpu
 
 # PyTorch image transformers
 timm==0.9.2