MIH-AI gpu enable

2025-05-12 10:57:06 +02:00 · 2025-05-12 10:57:06 +02:00 · 0291dd19d3
commit 0291dd19d3
parent c511fc4ea6
1 changed files with 9 additions and 8 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@ -169,13 +169,14 @@ services:
    networks:
      - MIH-network
    # === Added section for NVIDIA GPU acceleration ===
-    # deploy:
-    #   resources:
-    #     reservations:
-    #       devices:
-    #         - driver: nvidia
-    #           count: all # or specify a number of GPUs
-    #           capabilities: [gpu]
+    runtime: nvidia
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all # or specify a number of GPUs
+              capabilities: [ gpu ]
  #============== Firebaase ====================================================================   
  # firebase:
  #   container_name: MIH-firebase-emulator