diff --git a/.github/workflows/ci-pytorch.yml b/.github/workflows/ci-pytorch.yml index 1cea82cded..e94357e39a 100644 --- a/.github/workflows/ci-pytorch.yml +++ b/.github/workflows/ci-pytorch.yml @@ -28,24 +28,18 @@ jobs: fail-fast: false matrix: include: - - name: PyTorch 1.12.1 (Python 3.9) - framework: pytorch - python: 3.9 - torch: 1.12.1+cpu - torchvision: 0.13.1+cpu - torchaudio: 0.12.1 - - name: PyTorch 1.13.1 (Python 3.9) - framework: pytorch - python: 3.9 - torch: 1.13.1+cpu - torchvision: 0.14.1+cpu - torchaudio: 0.13.1 - name: PyTorch 1.13.1 (Python 3.10) framework: pytorch python: '3.10' torch: 1.13.1+cpu torchvision: 0.14.1+cpu torchaudio: 0.13.1 + - name: PyTorch 2.1.1 (Python 3.10) + framework: pytorch + python: '3.10' + torch: 2.1.1 + torchvision: 2.1.1 + torchaudio: 2.1.1 name: ${{ matrix.name }} steps: