diff --git a/docker-compose.yml b/docker-compose.yml
index acdffae..579ad1e 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,7 +11,13 @@ services:
     image: ollama/ollama:latest
     ports:
       - 11434:11434
-
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
   ollama-webui:
     image: ghcr.io/ollama-webui/ollama-webui:main
     container_name: ollama-webui