Merge pull request #779 from steveh250/main

Updated GPU resource reservations to docker-compose
2026-01-14 10:59:53 -05:00 · 2026-01-14 10:59:53 -05:00 · 99546cd0e7
commit 99546cd0e7
parent dfc97a161c 2b383ee325
1 changed files with 7 additions and 2 deletions
--- a/docker-compose.gpu.yml
+++ b/docker-compose.gpu.yml
@ -3,5 +3,10 @@ services:
    environment:
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - NVIDIA_VISIBLE_DEVICES=all
-    gpus: all
-
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]