From d3bd07dcc659cbf2efff2c4e290fae835cc7d678 Mon Sep 17 00:00:00 2001
From: Erik Stambaugh <erik@erikstambaugh.com>
Date: Sat, 2 Mar 2024 14:51:50 -0800
Subject: [PATCH] Update ulimits according to nvidia's recommendations

---
 docker-compose.yaml | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

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]
+