diff --git a/kerasDeloyment.yaml b/kerasDeloyment.yaml
index c452f36ad28e66c08b3880700f4cd0c70f7dc116..b50dbedcb7e52da4d672bfcbc5e4489b4cd93afc 100644
--- a/kerasDeloyment.yaml
+++ b/kerasDeloyment.yaml
@@ -1,17 +1,17 @@
 apiVersion: apps/v1
 kind: Deployment
 metadata:
-  name: keras
+  name: keras-gpu
   namespace: digits
 spec:
   replicas: 1
   selector:
     matchLabels:
-      k8s-app: keras
+      k8s-app: keras-gpu
   template:
     metadata:
       labels:
-        k8s-app: keras
+        k8s-app: keras-gpu
     spec:
       tolerations:
       - key: "region"
@@ -19,7 +19,7 @@ spec:
         value: "allow"
         effect: "NoSchedule"
       containers:
-      - name: keras
+      - name: keras-gpu
         image: gitlab-registry.nautilus.optiputer.net/ar-noc/keras-smoke-detection
         securityContext:
           capabilities:
@@ -44,12 +44,12 @@ spec:
           persistentVolumeClaim:
             claimName: modeltraining
             
-    #   affinity:
-    #     nodeAffinity:
-    #       requiredDuringSchedulingIgnoredDuringExecution:
-    #         nodeSelectorTerms:
-    #         - matchExpressions:
-    #           - key: gpu-type
-    #             operator: In # Use NotIn for other types
-    #             values:
-    #             - M40
+      affinity:
+        nodeAffinity:
+          requiredDuringSchedulingIgnoredDuringExecution:
+            nodeSelectorTerms:
+            - matchExpressions:
+              - key: gpu-type
+                operator: In # Use NotIn for other types
+                values:
+                - M40