Update ulimits according to nvidia's recommendations

2024-03-02 14:51:50 -08:00 · 2024-03-02 14:51:50 -08:00 · d3bd07dcc6
commit d3bd07dcc6
parent d1baec0f60
1 changed files with 7 additions and 3 deletions
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@ -22,13 +22,17 @@ services:
      - /dev/nvidia-modeset
      - /dev/nvidiactl
      - /dev/nvidia0
+    ulimits:
+      memlock: -1
+      stack: 67108864
+    ipc: host
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
-              count: 1
-              capabilities:
-                - gpu
+              count: all
+              capabilities: [gpu]
+