15 lines
289 B
YAML
15 lines
289 B
YAML
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
|