diff --git a/docs/gpu-sharing/gpu-memory.yaml b/docs/gpu-sharing/gpu-memory.yaml index cdabc80ab..74e4a37b5 100644 --- a/docs/gpu-sharing/gpu-memory.yaml +++ b/docs/gpu-sharing/gpu-memory.yaml @@ -12,6 +12,7 @@ metadata: spec: schedulerName: kai-scheduler containers: - - name: ubuntu - image: ubuntu - args: ["sleep", "infinity"] \ No newline at end of file + - name: gpu-workload + image: nvidia/cuda:13.0.2-base-ubi8 + command: ["nvidia-smi"] + args: ["-L"] \ No newline at end of file diff --git a/docs/gpu-sharing/gpu-sharing-init-container.yaml b/docs/gpu-sharing/gpu-sharing-init-container.yaml index f37cd4463..a3233dd9b 100644 --- a/docs/gpu-sharing/gpu-sharing-init-container.yaml +++ b/docs/gpu-sharing/gpu-sharing-init-container.yaml @@ -16,7 +16,7 @@ spec: schedulerName: kai-scheduler initContainers: - name: gpu-init - image: nvidia/cuda:11.0-base + image: nvidia/cuda:13.0.2-base-ubi8 command: ["nvidia-smi"] args: ["-L"] containers: diff --git a/docs/gpu-sharing/gpu-sharing-non-default-container.yaml b/docs/gpu-sharing/gpu-sharing-non-default-container.yaml index 1bab2d1ab..f3b7d696a 100644 --- a/docs/gpu-sharing/gpu-sharing-non-default-container.yaml +++ b/docs/gpu-sharing/gpu-sharing-non-default-container.yaml @@ -20,7 +20,7 @@ spec: image: busybox args: ["sleep", "infinity"] - name: gpu-workload - image: nvidia/cuda:11.0-base + image: nvidia/cuda:13.0.2-base-ubi8 command: ["nvidia-smi"] args: ["-L"] - name: another-sidecar diff --git a/docs/gpu-sharing/gpu-sharing.yaml b/docs/gpu-sharing/gpu-sharing.yaml index 2ece845da..31cd7846f 100644 --- a/docs/gpu-sharing/gpu-sharing.yaml +++ b/docs/gpu-sharing/gpu-sharing.yaml @@ -12,6 +12,7 @@ metadata: spec: schedulerName: kai-scheduler containers: - - name: ubuntu - image: ubuntu - args: ["sleep", "infinity"] \ No newline at end of file + - name: gpu-workload + image: nvidia/cuda:13.0.2-base-ubi8 + command: ["nvidia-smi"] + args: ["-L"] \ No newline at end of file diff --git a/docs/gpu-sharing/mps/gpu-sharing-with-mps.yaml b/docs/gpu-sharing/mps/gpu-sharing-with-mps.yaml index ae6b9882a..11d81ed7a 100644 --- a/docs/gpu-sharing/mps/gpu-sharing-with-mps.yaml +++ b/docs/gpu-sharing/mps/gpu-sharing-with-mps.yaml @@ -12,9 +12,10 @@ metadata: spec: schedulerName: kai-scheduler containers: - - name: ubuntu - image: ubuntu - args: ["sleep", "infinity"] + - name: gpu-workload + image: nvidia/cuda:13.0.2-base-ubi8 + command: ["nvidia-smi"] + args: ["-L"] volumeMounts: - name: mps mountPath: /tmp/nvidia-mps