try manual cuda installation

samayala22 · Aug 29, 2024 · 838957c · 838957c
1 parent bbb69e1
commit 838957c
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/.github/workflows/linux_gpu.yaml b/.github/workflows/linux_gpu.yaml
@@ -29,11 +29,17 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v4
 
-      - name: Install CUDA Toolkit
-        uses: Jimver/cuda-toolkit@v0.2.16
-        id: cuda-toolkit
-        with:
-          cuda: '12.4.0'
+      # - name: Install CUDA Toolkit
+      #   uses: Jimver/cuda-toolkit@v0.2.16
+      #   id: cuda-toolkit
+      #   with:
+      #     cuda: '12.4.0'
+      - name: Install CUDA Driver + Toolkit
+        run: |
+          wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
+          sudo dpkg -i cuda-keyring_1.1-1_all.deb
+          sudo apt-get update
+          sudo apt-get -y install cuda-toolkit-12-4 nvidia-open
 
       - name: Check CUDA info
         run: nvidia-smi