fix gpus

limit is required
machine-learning-apps · May 2, 2019 · 60234fd · 60234fd
1 parent c4fc197
commit 60234fd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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"]