docker-compose.yml

@@ -1,37 +1,29 @@
+version: '3.8'
+
 services:
   ollama:
     volumes:
-      - ./ollama:/root/.ollama
+      - $HOME/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]
 
-  open-webui:
-    image: ghcr.io/open-webui/open-webui:main
-    ports:
-      - "3000:8080"
-    environment:
-      - OLLAMA_BASE_URL=http://ollama:11434
+  ollama-webui:
+    image: ghcr.io/ollama-webui/ollama-webui:main
+    container_name: ollama-webui
     volumes:
-      - ./open-webui:/app/backend/data
-    container_name: open-webui
-    restart: unless-stopped # Optional: Add a restart policy
+      - $HOME/ollama/ollama-webui:/app/backend/data
+    depends_on:
+      - ollama
+    ports:
+      - 3000:8080
+    environment:
+      - '/ollama/api=http://ollama:11434/api'
+    extra_hosts:
+      - host.docker.internal:host-gateway
+    restart: unless-stopped
+