diff --git a/ollama37/docker-compose.yml b/ollama37/docker-compose.yml
index 9491bfb..7b61afd 100644
--- a/ollama37/docker-compose.yml
+++ b/ollama37/docker-compose.yml
@@ -9,7 +9,7 @@ services:
     volumes:
       - ollama:/root/.ollama
     restart: unless-stopped
-
+    runtime: nvidia
 volumes:
   ollama: