feat(nvidia-device-plugin): install nvidia-device-plugin
This commit is contained in:
14
nvidia-device-plugin/gpu-test-pod.yaml
Normal file
14
nvidia-device-plugin/gpu-test-pod.yaml
Normal file
@@ -0,0 +1,14 @@
|
||||
apiVersion: v1
|
||||
kind: Pod
|
||||
metadata:
|
||||
name: gpu-test
|
||||
spec:
|
||||
restartPolicy: OnFailure
|
||||
runtimeClassName: nvidia
|
||||
containers:
|
||||
- name: cuda-container
|
||||
image: nvcr.io/nvidia/cuda:12.2.0-base-ubuntu22.04
|
||||
command: ["nvidia-smi"]
|
||||
resources:
|
||||
limits:
|
||||
nvidia.com/gpu: 1
|
||||
Reference in New Issue
Block a user