diff --git a/docker-compose.yaml b/docker-compose.yaml
index 77179c7..109fb35 100644
--- 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]
+