diff --git a/deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml b/deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml index c80b5112c..6bc4b5d2b 100644 --- a/deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml +++ b/deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml @@ -200,7 +200,7 @@ spec: value: {{ .Values.deviceDiscoveryStrategy }} {{- end }} - name: NVIDIA_VISIBLE_DEVICES - value: all + value: {{ .Values.visibleDevices }} - name: NVIDIA_DRIVER_CAPABILITIES value: compute,utility securityContext: diff --git a/deployments/helm/nvidia-device-plugin/values.yaml b/deployments/helm/nvidia-device-plugin/values.yaml index 12d30964a..278bc83cf 100644 --- a/deployments/helm/nvidia-device-plugin/values.yaml +++ b/deployments/helm/nvidia-device-plugin/values.yaml @@ -37,6 +37,7 @@ gdrcopyEnabled: null gdsEnabled: null mofedEnabled: null deviceDiscoveryStrategy: null +visibleDevices: all nameOverride: "" fullnameOverride: ""