diff --git a/charts/gpu-metrics-exporter/templates/daemonset.yaml b/charts/gpu-metrics-exporter/templates/daemonset.yaml index d563b72..b312a5e 100644 --- a/charts/gpu-metrics-exporter/templates/daemonset.yaml +++ b/charts/gpu-metrics-exporter/templates/daemonset.yaml @@ -53,6 +53,10 @@ spec: {{- toYaml . | nindent 8 }} {{- end }} {{- end}} + {{- with .Values.gpuMetricsExporter.tolerations }} + tolerations: + {{- toYaml . | nindent 8 }} + {{- end }} containers: - name: castai-gpu-metrics-exporter securityContext: diff --git a/charts/gpu-metrics-exporter/values.yaml b/charts/gpu-metrics-exporter/values.yaml index 70cc36a..97904ec 100644 --- a/charts/gpu-metrics-exporter/values.yaml +++ b/charts/gpu-metrics-exporter/values.yaml @@ -27,6 +27,10 @@ gpuMetricsExporter: readOnlyRootFilesystem: true runAsNonRoot: true port: 6061 + tolerations: + - key: "nvidia.com/gpu" + value: "true" + effect: "NoSchedule" dcgmExporter: enabled: true