Update test-cdk-changes.yaml

Author: Jyothirmaikottu

.github/workflows/test-cdk-changes.yaml

@@ -15,8 +15,10 @@ jobs:
 
       - name: Check Docker
         run: |
+          docker info
+          echo "Pod NVIDIA_VISIBLE_DEVICES: $NVIDIA_VISIBLE_DEVICES"
           docker pull nvidia/cuda:12.0.0-base-ubuntu22.04
-          docker run --rm --gpus all nvidia/cuda:12.0.0-base-ubuntu22.04 nvidia-smi
+          docker run --rm --gpus "\"device=${NVIDIA_VISIBLE_DEVICES}\"" nvidia/cuda:12.0.0-base-ubuntu22.04 nvidia-smi
 
   test-p4d:
     runs-on: p4d-gpu-runners
@@ -26,5 +28,7 @@ jobs:
 
       - name: Check Docker
         run: |
+          docker info
+          echo "Pod NVIDIA_VISIBLE_DEVICES: $NVIDIA_VISIBLE_DEVICES"
           docker pull nvidia/cuda:12.0.0-base-ubuntu22.04
-          docker run --rm --gpus all nvidia/cuda:12.0.0-base-ubuntu22.04 nvidia-smi
+          docker run --rm --gpus "\"device=${NVIDIA_VISIBLE_DEVICES}\"" nvidia/cuda:12.0.0-base-ubuntu22.04 nvidia-smi