Update test-native-gpu-runner.yaml

Author: Jyothirmaikottu

.github/workflows/test-native-gpu-runner.yaml

@@ -80,3 +80,31 @@ jobs:
           sleep 60
           echo "Finished workload at: $(date -u +%Y-%m-%dT%H:%M:%S)"
           echo "✅ Job ${{ matrix.job_id }} completed"
+
+  # Test Docker container with GPU access
+  test-docker-gpu:
+    runs-on: g6-2gpu-native-runner
+    steps:
+      - name: Check Docker
+        run: |
+          echo "=== Docker Info ==="
+          docker version
+          docker info | grep -i runtime
+
+      - name: Run GPU Container
+        run: |
+          echo "=== Running CUDA container ==="
+          docker run --rm --gpus all nvidia/cuda:12.2.0-base-ubuntu22.04 nvidia-smi
+
+      - name: Run PyTorch GPU Test
+        run: |
+          echo "=== Running PyTorch GPU test ==="
+          docker run --rm --gpus all pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime \
+            python -c "
+          import torch
+          print(f'PyTorch version: {torch.__version__}')
+          print(f'CUDA available: {torch.cuda.is_available()}')
+          print(f'GPU count: {torch.cuda.device_count()}')
+          for i in range(torch.cuda.device_count()):
+              print(f'GPU {i}: {torch.cuda.get_device_name(i)}')
+          "