add cuda to CI

faberno · Oct 3, 2024 · e477c34 · e477c34
1 parent 19fe49e
commit e477c34
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -40,6 +40,13 @@ jobs:
         os: [ubuntu-latest, windows-latest]
 
     steps:
+    - name: Fetch CUDA toolkit
+      uses: Jimver/[email protected]
+      id: cuda-toolkit
+      with:
+        cuda: '12.5.0'
+        linux-local-args: '["--toolkit"]'
+
     - uses: actions/checkout@v4
       with:
         submodules: true