switch to cuda base image

Signed-off-by: Swati Gupta <[email protected]>

Author: guptaNswati

templates/compute-domain-daemon.tmpl.yaml

@@ -22,7 +22,7 @@ spec:
         {{ .ComputeDomainLabelKey }}: {{ .ComputeDomainLabelValue }}
       containers:
       - name: compute-domain-daemon
-        image: ubuntu:22.04
+        image: nvcr.io/nvidia/cuda:12.2.0-base-ubuntu22.04
         command: [sh, -c]
         args:
         - |-

templates/mps-control-daemon.tmpl.yaml

@@ -20,7 +20,7 @@ spec:
       hostPID: true
       containers:
       - name: mps-control-daemon
-        image: ubuntu:22.04
+        image: nvcr.io/nvidia/cuda:12.2.0-base-ubuntu22.04
         securityContext:
           privileged: true
         command: [chroot, /driver-root, sh, -c]