Update test-g6-non-dind-runners.yaml

Author: Jyothirmaikottu

.github/workflows/test-g6-non-dind-runners.yaml

@@ -1,4 +1,3 @@
-
 name: Test - Multi-GPU Non DIND Runners
 
 on:
@@ -8,31 +7,37 @@ on:
       - ".github/workflows/test-g6-non-dind-runners.yaml"
 
 jobs:
-  test-1gpu-runner:
+  test-2gpu-runner:
     runs-on: g6-non-dind-runner
+    container:
+      image: nvidia/cuda:12.2.0-runtime-ubuntu22.04
     steps:
       - name: Checkout
         uses: actions/checkout@v4
 
       - name: Job Info
         run: |
-          echo "=== 1-GPU Runner Test ==="
+          echo "=== 2-GPU Runner Test ==="
           echo "Runner: $(hostname)"
           echo "Timestamp: $(date -u)"
+
+      - name: Check GPU Info
+        run: |
+          echo "=== GPU Information ==="
+          nvidia-smi
+          echo ""
+          echo "GPU Count: $(nvidia-smi -L | wc -l)"
+          nvidia-smi --query-gpu=index,name,memory.total,memory.used --format=csv
+
       - name: Check Node Resources
         run: |
           echo "=== Node Information ==="
           echo "Hostname: $(hostname)"
           echo "CPU cores: $(nproc)"
           echo "Memory: $(free -h | grep Mem | awk '{print $2}')"
-      - name: Run GPU Test
+
+      - name: Run CUDA Test
         run: |
-          docker run --rm --gpus=all \
-            nvidia/cuda:12.2.0-base-ubuntu22.04 \
-            bash -c "
-              echo '=== 1-GPU Test ==='
-              nvidia-smi
-              echo ''
-              echo '✅ 1-GPU test completed'
-            "
-  
+          echo "=== CUDA Version ==="
+          nvcc --version || echo "nvcc not installed in runtime image"
+          cat /usr/local/cuda/version.txt 2>/dev/null || echo "CUDA version file not found"