Update PyTorch workflow for torch

Signed-off-by: Beat Buesser <[email protected]>
Trusted-AI · Dec 20, 2023 · 8b89f11 · 8b89f11
1 parent 65925a4
commit 8b89f11
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/ci-pytorch.yml b/.github/workflows/ci-pytorch.yml
@@ -34,12 +34,12 @@ jobs:
             torch: 1.13.1+cpu
             torchvision: 0.14.1+cpu
             torchaudio: 0.13.1
-          - name: PyTorch 2.1.1 (Python 3.10)
+          - name: PyTorch 2.1.2 (Python 3.10)
             framework: pytorch
             python: '3.10'
-            torch: 2.1.1
-            torchvision: 0.16.1+cpu
-            torchaudio: 2.1.1
+            torch: 2.1.2
+            torchvision: 0.16.2+cpu
+            torchaudio: 2.1.2
 
     name: ${{ matrix.name }}
     steps:

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