updated

docker-compose.yml

@@ -1,29 +1,37 @@
-version: '3.8'
-
 services:
   ollama:
     volumes:
-      - $HOME/ollama/ollama:/root/.ollama
+      - ./ollama:/root/.ollama
     container_name: ollama
     pull_policy: always
     tty: true
     restart: unless-stopped
     image: ollama/ollama:latest
+    healthcheck:
+      test: ["CMD-SHELL", "nvidia-smi > /dev/null"]
+      interval: 10s
+      timeout: 30s
+      retries: 3
+    environment:
+      - OLLAMA_KEEP_ALIVE=60m
+      - OLLAMA_NUM_PARALLEL=4
     ports:
       - 11434:11434
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
 
-  ollama-webui:
+  open-webui:
-    image: ghcr.io/ollama-webui/ollama-webui:main
+    image: ghcr.io/open-webui/open-webui:main
-    container_name: ollama-webui
-    volumes:
-      - $HOME/ollama/ollama-webui:/app/backend/data
-    depends_on:
-      - ollama
     ports:
-      - 3000:8080
+      - "3000:8080"
     environment:
-      - '/ollama/api=http://ollama:11434/api'
+      - OLLAMA_BASE_URL=http://ollama:11434
-    extra_hosts:
+    volumes:
-      - host.docker.internal:host-gateway
+      - ./open-webui:/app/backend/data
-    restart: unless-stopped
+    container_name: open-webui
-
+    restart: unless-stopped # Optional: Add a restart policy