42 lines
747 B
YAML
42 lines
747 B
YAML
---
|
|
|
|
apiVersion: v1
|
|
kind: Namespace
|
|
metadata:
|
|
name: gpu-operator
|
|
|
|
---
|
|
|
|
apiVersion: helm.cattle.io/v1
|
|
kind: HelmChart
|
|
metadata:
|
|
name: gpu-operator
|
|
namespace: gpu-operator
|
|
spec:
|
|
repo: https://helm.ngc.nvidia.com/nvidia
|
|
chart: gpu-operator
|
|
targetNamespace: gpu-operator
|
|
version: 24.9.2
|
|
|
|
---
|
|
|
|
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 |