39 lines
982 B
YAML
39 lines
982 B
YAML
apiVersion: v1
|
|
kind: Namespace
|
|
metadata:
|
|
name: gpu-operator
|
|
---
|
|
apiVersion: apps/v1
|
|
kind: DaemonSet
|
|
metadata:
|
|
name: nvidia-device-plugin-daemonset
|
|
namespace: gpu-operator
|
|
spec:
|
|
selector:
|
|
matchLabels:
|
|
name: nvidia-device-plugin-ds
|
|
template:
|
|
metadata:
|
|
labels:
|
|
name: nvidia-device-plugin-ds
|
|
spec:
|
|
tolerations:
|
|
- key: nvidia.com/gpu
|
|
operator: Exists
|
|
effect: NoSchedule
|
|
containers:
|
|
- image: nvcr.io/nvidia/k8s-device-plugin:v0.14.0
|
|
name: nvidia-device-plugin-ctr
|
|
args: ["--fail-on-init-error=false"]
|
|
env:
|
|
- name: NVIDIA_VISIBLE_DEVICES
|
|
value: "all"
|
|
- name: MIG_STRATEGY
|
|
value: "single"
|
|
volumeMounts:
|
|
- name: device-plugin
|
|
mountPath: /var/lib/kubelet/device-plugins
|
|
volumes:
|
|
- name: device-plugin
|
|
hostPath:
|
|
path: /var/lib/kubelet/device-plugins |