2024-04-06 08:11:33 +08:00
|
|
|
services:
|
|
|
|
|
ollama:
|
|
|
|
|
devices:
|
|
|
|
|
- /dev/kfd:/dev/kfd
|
|
|
|
|
- /dev/dri:/dev/dri
|
2024-04-06 11:46:20 +08:00
|
|
|
image: ollama/ollama:${OLLAMA_DOCKER_TAG-rocm}
|
2024-04-06 08:11:33 +08:00
|
|
|
environment:
|
2024-04-06 11:46:20 +08:00
|
|
|
- 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION-11.0.0}'
|
