version: '3.8'
services:
ollama:
image: ollama/ollama:rocm # 使用 ROCm 专用镜像
container_name: ollama
environment:
- HSA_OVERRIDE_GFX_VERSION=11.0.0 # 强制识别为 gfx1100
- HIP_VISIBLE_DEVICES=0
- OLLAMA_GPU_LAYERS=all
- HSA_ENABLE_SDMA=0
devices:
- "/dev/kfd:/dev/kfd"
- "/dev/dri:/dev/dri"
- "/dev/dri/renderD128:/dev/dri/renderD128"
ports:
- "11434:11434" # API 端口
volumes:
- "/vol1/1000/docker/ollama/models:/root/.ollama"
参考上述代码,修改自己的volumes挂载