diff --git a/kerasDeloyment.yaml b/kerasDeloyment.yaml
index 2cda0a1d5e608a8bb720fb15fee27e5b66336c69..4dbbf5a9460a01dced2732aa5caaa6dea3739fac 100644
--- a/kerasDeloyment.yaml
+++ b/kerasDeloyment.yaml
@@ -35,7 +35,16 @@ spec:
           requests:
             memory: "32Gi"
             cpu: "4"
-    
+      affinity:
+        nodeAffinity:
+          requiredDuringSchedulingIgnoredDuringExecution:
+            nodeSelectorTerms:
+            - matchExpressions:
+              - key: gpu-type
+                operator: In # Use NotIn for other types
+                values:
+                - M40
+
         volumeMounts:
         - mountPath: /userdata/kerasData
           name: modeltraining