Add workflow for Pytorch 2.x

Signed-off-by: Beat Buesser <[email protected]>
Trusted-AI · Nov 30, 2023 · 6e32356 · 6e32356
1 parent 81e6232
commit 6e32356
Showing 1 changed file with 6 additions and 12 deletions.
diff --git 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: