Unverified Commit 1d4dcf8e authored by XiaoLiang Hu's avatar XiaoLiang Hu Committed by GitHub
Browse files

fix(installer): enable-nvidia-device (#676)

parent df92a5c4
Showing with 21 additions and 0 deletions
+21 -0
......@@ -62,6 +62,19 @@ spec:
nodeSelector:
nvidia-device-enable: enable
hostPID: true
initContainers:
- name: nvidia-uvm-enable
image: {{ .BusyboxImage }}
command: ["sh", "-c"," /sbin/modprobe nvidia-uvm 2>/dev/null && mknod -m 666 /dev/nvidia-uvm c `grep nvidia-uvm /proc/devices | awk '{print $1}'` 0 2>/dev/null || echo 0"]
imagePullPolicy: IfNotPresent
securityContext:
privileged: true
volumeMounts:
- mountPath: /lib/modules
name: lib-modules
readOnly: true
- mountPath: /dev
name: dev
containers:
- name: gpu-manager
image: {{ .GPUManagerImage }}
......@@ -103,6 +116,14 @@ spec:
fieldRef:
fieldPath: spec.nodeName
volumes:
- name: lib-modules
hostPath:
type: Directory
path: /lib/modules
- name: dev
hostPath:
type: Directory
path: /dev
- name: device-plugin
hostPath:
type: Directory
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment