GPU driver check

gadgetron · Nov 13, 2024 · 8d15fb1 · 8d15fb1
1 parent 6f227ae
commit 8d15fb1
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -53,4 +53,10 @@ jobs:
 
       - name: GPU Check
         shell: bash
-        run: nvidia-smi
+        run: |
+          python3 -V
+          apt update
+          apt install ubuntu-drivers-common
+          ubuntu-drivers list --gpgpu
+          ubuntu-drivers install --gpgpu
+          nvidia-smi