update tested CUDA versions in CI

pmeier · Dec 8, 2024 · 212b91e · 212b91e
1 parent 2eeed03
commit 212b91e
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/.github/workflows/install.yml b/.github/workflows/install.yml
@@ -71,22 +71,24 @@ jobs:
           - windows-latest
           - macos-latest
         python-version:
-          - "3.8"
           - "3.9"
           - "3.10"
           - "3.11"
         pytorch-computation-backend:
           - cpu
-          - cu117
           - cu118
+          - cu121
+          - cu124
         exclude:
-          - os: macos-latest
-            pytorch-computation-backend: cu117
           - os: macos-latest
             pytorch-computation-backend: cu118
+          - os: macos-latest
+            pytorch-computation-backend: cu121
+          - os: macos-latest
+            pytorch-computation-backend: cu124
       # TODO: find a way to test this
       #          - os: ubuntu-latest
-      #            pytorch-computation-backend: rocm5.4.2
+      #            pytorch-computation-backend: rocm6.2
       fail-fast: false
 
     runs-on: ${{ matrix.os }}
@@ -108,7 +110,7 @@ jobs:
       - name: Install torch
         run:
           ltt install --pytorch-computation-backend=${{
-          matrix.pytorch-computation-backend }} torch==2.0.0
+          matrix.pytorch-computation-backend }} torch==2.5.1
 
       - name: Check computation backend
         shell: python