diff --git a/deployments/gpu-operator/templates/cleanup_crd.yaml b/deployments/gpu-operator/templates/cleanup_crd.yaml index 0d426f952..4029a9988 100644 --- a/deployments/gpu-operator/templates/cleanup_crd.yaml +++ b/deployments/gpu-operator/templates/cleanup_crd.yaml @@ -30,6 +30,13 @@ spec: tolerations: {{- toYaml . | nindent 8 }} {{- end }} + {{- if .Values.operator.nodeSelector }} + nodeSelector: + {{- toYaml .Values.operator.nodeSelector | nindent 8 }} + {{- else }} + nodeSelector: + kubernetes.io/os: linux + {{- end }} containers: - name: cleanup-crd image: {{ include "gpu-operator.fullimage" . }} diff --git a/deployments/gpu-operator/templates/upgrade_crd.yaml b/deployments/gpu-operator/templates/upgrade_crd.yaml index e887b3a81..5ce412307 100644 --- a/deployments/gpu-operator/templates/upgrade_crd.yaml +++ b/deployments/gpu-operator/templates/upgrade_crd.yaml @@ -79,6 +79,12 @@ spec: tolerations: {{- toYaml . | nindent 8 }} {{- end }} + nodeSelector: + {{- if .Values.operator.nodeSelector }} + {{- toYaml .Values.operator.nodeSelector | nindent 8 }} + {{- else }} + kubernetes.io/os: linux + {{- end }} containers: - name: upgrade-crd image: {{ include "gpu-operator.fullimage" . }}