diff --git a/deployment/dcgm-exporter/templates/daemonset.yaml b/deployment/dcgm-exporter/templates/daemonset.yaml
index b32bd2d..3329527 100644
--- a/deployment/dcgm-exporter/templates/daemonset.yaml
+++ b/deployment/dcgm-exporter/templates/daemonset.yaml
@@ -97,7 +97,10 @@ spec:
           initialDelaySeconds: 5
         resources:
           {{- toYaml .Values.resources | nindent 12 }}
-
+    {{- with .Values.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+    {{- end }}
     {{- with .Values.tolerations }}
       tolerations:
         {{- toYaml . | nindent 8 }}