MIH-AI gpu enable

2025-05-12 10:57:06 +02:00
parent c511fc4ea6
commit 0291dd19d3
1 changed files with 9 additions and 8 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -168,14 +168,15 @@ services:
      - OLLAMA_HOST=0.0.0.0
    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]
+    # === Added section for NVIDIA GPU acceleration ===
+    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