diff --git a/argo/workflow.yaml b/argo/workflow.yaml
index 8cb21b61..11fe82a2 100644
--- a/argo/workflow.yaml
+++ b/argo/workflow.yaml
@@ -49,4 +49,6 @@ spec:
       resources:
         requests:
           nvidia.com/gpu: 1 # requesting 1 GPU
+        limits:
+          nvidia.com/gpu: 1
       command: ["python", "/src/train.py"]