install driver then reboot

samayala22 · Aug 29, 2024 · c09856a · c09856a
1 parent 838957c
commit c09856a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/workflows/linux_gpu.yaml b/.github/workflows/linux_gpu.yaml
@@ -39,10 +39,10 @@ jobs:
           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
+          sudo apt-get -y install nvidia-open
+          sudo reboot
+          sudo apt-get -y install cuda-toolkit-12-4
+          nvidia-smi
 
       - name: Download ISPC
         run: |