diff --git a/docker-compose.yml b/docker-compose.yml
index c45933e..39e1edf 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -3,6 +3,14 @@ services:
         volumes:
         - ./files/ollama:/root/.ollama
         container_name: ollama
+        #UNCOMMENT THESE LINES IF YOU HAVE AN NVIDIA GPU
+        #deploy:
+        #    resources:
+        #        reservations:
+        #            devices:
+        #                - driver: nvidia
+        #                  count: all
+        #                  capabilities: [gpu]
         pull_policy: always
         tty: true
         restart: unless-stopped