From 0291dd19d33355e836514e0229eae2ff69183bb2 Mon Sep 17 00:00:00 2001
From: yaso <yasienmeth@gmail.com>
Date: Mon, 12 May 2025 10:57:06 +0200
Subject: [PATCH] MIH-AI gpu enable

---
 docker-compose.yml | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index c6cae1fd..d7317cdb 100644
--- 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