support CUDA

intelligent-control-lab · Jun 20, 2024 · 9709be0 · 9709be0
1 parent 283a908
commit 9709be0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -21,8 +21,6 @@ jobs:
           - '1.9.3'
         os:
           - ubuntu-latest
-          - macOS-latest
-          - windows-latest
         arch:
           - x64
     steps:
@@ -41,10 +39,12 @@ jobs:
             ${{ runner.os }}-test-${{ env.cache-name }}-
             ${{ runner.os }}-test-
             ${{ runner.os }}-
-      - uses: Jimver/[email protected]
-        id: cuda-toolkit
-        with:
-          cuda: '12.4.1'
+      - name: Set up CUDA
+        if: matrix.os == 'ubuntu-latest'
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y nvidia-cuda-toolkit
+          sudo apt-get install -y nvidia-driver-470
       - name: Install local dependencies
         run: |
           julia --project=. -e 'using Pkg; Pkg.develop(path="onnx_parser/NaiveNASflux")'