ci: fix cuda version for wheel build workflow (#282)

vectorch-ai · Jul 25, 2024 · 2e14170 · 2e14170
1 parent 9505a50
commit 2e14170
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/.github/workflows/build_wheel.yml b/.github/workflows/build_wheel.yml
@@ -24,11 +24,13 @@ jobs:
       fail-fast: false
       matrix:
         python: ["3.8", "3.9", "3.10", "3.11", "3.12"]
-        cuda: ["11.8", "12.1"]
+        cuda: ["11.8", "12.1", "12.4"]
         torch: ["2.2.2", "2.3.1", "2.4.0"]
-        include:
+        exclude:
           - cuda: "12.4"
-            torch: "2.4.0"
+            torch: "2.3.1"
+          - cuda: "12.4"
+            torch: "2.2.2"
     runs-on: [self-hosted, linux, release]
     env:
       PYTHON_VERSION: ${{ matrix.python }}