Added nvidia-smi.yaml

saicharithpasula · Jul 23, 2024 · 9c19322 · 9c19322
1 parent 3dbc5de
commit 9c19322
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/nvidia-smi.yaml b/test/nvidia-smi.yaml
@@ -0,0 +1,14 @@
+apiVersion: v1 
+kind: Pod 
+metadata: 
+  name: nvidia-smi 
+spec: 
+  restartPolicy: OnFailure 
+  runtimeClassName: nvidia 
+  containers: 
+    - name: nvidia-smi 
+      image: "nvidia/cuda:12.4.0-base-ubuntu22.04" 
+      args: ["nvidia-smi"] 
+      resources: 
+         limits: 
+           nvidia.com/gpu: 1