20 lines
444 B
YAML
20 lines
444 B
YAML
apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
name: nbody-gpu-benchmark
|
|
namespace: default
|
|
spec:
|
|
restartPolicy: OnFailure
|
|
runtimeClassName: nvidia
|
|
containers:
|
|
- name: cuda-container
|
|
image: nvcr.io/nvidia/k8s/cuda-sample:nbody
|
|
args: ["nbody", "-gpu", "-benchmark"]
|
|
resources:
|
|
limits:
|
|
nvidia.com/gpu: 1
|
|
env:
|
|
- name: NVIDIA_VISIBLE_DEVICES
|
|
value: all
|
|
- name: NVIDIA_DRIVER_CAPABILITIES
|
|
value: all |