Skip to content

Commit b345646

Browse files
committed
Update PyTorch workflow for torch
Signed-off-by: Beat Buesser <[email protected]>
1 parent 6cbe871 commit b345646

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

.github/workflows/ci-pytorch.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,12 +34,12 @@ jobs:
3434
torch: 1.13.1+cpu
3535
torchvision: 0.14.1+cpu
3636
torchaudio: 0.13.1
37-
- name: PyTorch 2.1.2 (Python 3.10)
37+
- name: PyTorch 2.2.1 (Python 3.10)
3838
framework: pytorch
3939
python: '3.10'
40-
torch: 2.1.2
41-
torchvision: 0.16.2+cpu
42-
torchaudio: 2.1.2
40+
torch: 2.2.1
41+
torchvision: 0.17.1+cpu
42+
torchaudio: 2.2.1
4343

4444
name: ${{ matrix.name }}
4545
steps:

requirements_test.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ mxnet-native==1.8.0.post0
3232
# PyTorch
3333
--find-links https://download.pytorch.org/whl/cpu/torch_stable.html
3434
torch==2.2.1
35-
torchaudio==2.1.2
36-
torchvision==0.16.2+cpu
35+
torchaudio==2.2.1
36+
torchvision==0.17.1+cpu
3737

3838
# PyTorch image transformers
3939
timm==0.9.2

0 commit comments

Comments
 (0)