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" . }}