Files
docker-compose-cookbooks/ollama/compose-gpu.yaml
T

25 lines
678 B
YAML

# Ollama GPU
services:
ollama:
image: ollama/ollama
volumes:
- ${VOL_PATH}/ollama:/root/.ollama
runtime: nvidia
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
ports:
- "11434:11434"
webui:
image: ${WEBUI_IMAGE:-ghcr.io/open-webui/open-webui:main}
volumes:
- "${VOL_PATH:-/data}/webui:/root/.cache"
- "${VOL_PATH:-/data}/open-webui:/app/backend/data"
environment:
- WEBUI_URL=${WEBUI_URL:-http://localhost:8080}
- WEBUI_SECRET_KEY=${WEBUI_SECRET_KEY:?must_be_set}
- OLLAMA_BASE_URL=http://ollama:11434
ports:
- "${WEBUI_PORT:-8080}:8080"