diff --git a/ollama/docker-compose.yml b/ollama/docker-compose.yml
index 5d5f07d..62a3911 100644
--- a/ollama/docker-compose.yml
+++ b/ollama/docker-compose.yml
@@ -8,7 +8,16 @@ services:
     restart: unless-stopped
     image: ollama/ollama:${OLLAMA_DOCKER_TAG-latest}
     ports:
-      - 11434:11434   
+      - 11434:11434
+    devices:
+      - /dev/dri:/dev/dri  
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: ["gpu"]
+              count: all
+
 
   open-webui:
     build:
@@ -27,10 +36,12 @@ services:
     environment:
       - 'OLLAMA_BASE_URL=http://ollama:11434'
       - 'WEBUI_SECRET_KEY='
+    devices:
+      - /dev/dri:/dev/dri    
     extra_hosts:
       - host.docker.internal:host-gateway
     restart: unless-stopped
 
 volumes:
   ollama: {}
-  open-webui: {}
\ No newline at end of file
+  open-webui: {}