Update test-cdk-changes.yaml

Author: Jyothirmaikottu

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

@@ -21,12 +21,8 @@ jobs:
 
   test-p4d:
     runs-on: p4d-gpu-runners
+    container:
+      image: nvidia/cuda:12.0.0-runtime-ubuntu22.04
     steps:
       - name: Check GPUs
         run: nvidia-smi
-
-      - name: Check Docker
-        run: |
-          docker info
-          docker pull nvidia/cuda:12.0.0-runtime-ubuntu22.04
-          docker run --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all nvidia/cuda:12.0.0-runtime-ubuntu22.04 nvidia-smi