replace the custom cuda github action by a docker container

Serapieum-of-alex · Aug 11, 2024 · 9fbf784 · 9fbf784
1 parent f554ae7
commit 9fbf784
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -11,6 +11,8 @@ jobs:
         python-version: ["3.12"]
     env:
       OS: ${{ matrix.os }}
+    container:
+      image: nvidia/cuda:12.4-base
 
     steps:
       - uses: actions/checkout@v3
@@ -20,10 +22,10 @@ jobs:
           python-version: ${{ matrix.python-version }}
           architecture: x64
 
-      - name: Install CUDA Toolkit
-        uses: Jimver/[email protected]
-        with:
-          cuda: "12.4.0"
+#       - name: Install CUDA Toolkit
+#         uses: Jimver/[email protected]
+#         with:
+#           cuda: "12.4.0"
 
       - name: Cleanup after cuda
         run: |