check cuda available #3

sensein · Sep 17, 2024 · 9fe0c97 · 9fe0c97
1 parent d5d1596
commit 9fe0c97
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.github/workflows/gpu_tests.yaml b/.github/workflows/gpu_tests.yaml
@@ -42,9 +42,11 @@ jobs:
       - uses: actions/checkout@v4
         with: 
           fetch-depth: 1
-      - name: Check for NVIDIA GPU
+      - name: Display NVIDIA SMI details
         run: |
-          nvidia-smi || echo "No NVIDIA GPU available!"
+          nvidia-smi
+          nvidia-smi -L
+          nvidia-smi -q -d Memory
       - name: Set up Python 3.10
         uses: actions/setup-python@v5
         with: