diff --git a/.github/workflows/ci-pytorch.yml b/.github/workflows/ci-pytorch.yml index 7bd53fdf60..2f8d9a4810 100644 --- 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 index fb56b039f9..f20eb231c5 100644 --- 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