Limit 1 gpu

pytorch · Jan 26, 2024 · 274fe04 · 274fe04
1 parent 396b522
commit 274fe04
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docker/infra/values.yaml b/docker/infra/values.yaml
@@ -241,8 +241,10 @@ template:
         - name: dshm
           mountPath: /dev/shm
       resources:
-        limits:
+        requests:
           nvidia.com/gpu: 1 # requesting 1 GPU
+        limits:
+          nvidia.com/gpu: 1 # limiting 1 GPU
     volumes:
     - name: nvidia-lib
       hostPath: