@@ -3,6 +3,6 @@ services:
devices:
- /dev/kfd:/dev/kfd
- /dev/dri:/dev/dri
- image: ollama/ollama:rocm
+ image: ollama/ollama:${OLLAMA_DOCKER_TAG-rocm}
environment:
- - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION}'
+ - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION-11.0.0}'