Seth
adeda6dd84
Ollama ROCm doesn't auto-detect newer AMD iGPUs (gfx1150/1151). Setting HSA_OVERRIDE_GFX_VERSION=11.0.0 in the compose fixes this. Configurable via .env for other AMD chips. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
64 lines
1.7 KiB
YAML
64 lines
1.7 KiB
YAML
version: "3.8"
|
|
|
|
# Mortdecai Inference Gateway
|
|
# Deploy on any machine with Ollama-compatible GPU
|
|
#
|
|
# Usage:
|
|
# docker compose up -d
|
|
# # Dashboard at http://localhost:8434/dashboard
|
|
#
|
|
# For AMD ROCm (Strix Halo, RX 7000, etc):
|
|
# Ollama image auto-detects ROCm. Ensure rocm drivers are installed on host.
|
|
#
|
|
# For NVIDIA:
|
|
# Requires nvidia-container-toolkit installed on host.
|
|
|
|
services:
|
|
ollama:
|
|
image: ollama/ollama:rocm
|
|
container_name: mortdecai-ollama
|
|
restart: unless-stopped
|
|
ports:
|
|
- "127.0.0.1:11434:11434" # Only accessible to gateway, not exposed
|
|
volumes:
|
|
- ollama-data:/root/.ollama
|
|
- ./models:/models
|
|
devices:
|
|
- /dev/kfd:/dev/kfd
|
|
- /dev/dri:/dev/dri
|
|
environment:
|
|
- OLLAMA_HOST=0.0.0.0:11434
|
|
- HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION:-11.0.0}
|
|
# For NVIDIA, replace 'devices' above with:
|
|
# deploy:
|
|
# resources:
|
|
# reservations:
|
|
# devices:
|
|
# - driver: nvidia
|
|
# count: all
|
|
# capabilities: [gpu]
|
|
|
|
gateway:
|
|
build: .
|
|
container_name: mortdecai-gateway
|
|
restart: unless-stopped
|
|
ports:
|
|
- "8434:8434" # This is the only exposed port
|
|
environment:
|
|
- OLLAMA_URL=http://ollama:11434
|
|
- API_KEY=${API_KEY:-mk_mortdecai_default}
|
|
- GATEWAY_PORT=8434
|
|
- GPU_TDP_WATTS=${GPU_TDP_WATTS:-54}
|
|
- SYSTEM_OVERHEAD_WATTS=${SYSTEM_OVERHEAD_WATTS:-30}
|
|
- ELECTRICITY_RATE=${ELECTRICITY_RATE:-0.15}
|
|
- SPENDING_CAP=${SPENDING_CAP:-10.00}
|
|
- STATS_FILE=/data/stats.json
|
|
volumes:
|
|
- gateway-data:/data
|
|
depends_on:
|
|
- ollama
|
|
|
|
volumes:
|
|
ollama-data:
|
|
gateway-data:
|