From 9709be05d01695406bb3d71611837f5159b34287 Mon Sep 17 00:00:00 2001
From: Tianhao <phi.wth@gmail.com>
Date: Thu, 20 Jun 2024 13:42:11 -0400
Subject: [PATCH] support CUDA

---
 .github/workflows/CI.yml | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index fb5b47c..fd310ba 100644
--- 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/cuda-toolkit@v0.2.15
-        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")'