Add variables

This commit is contained in:
Justin Hayes 2024-04-05 23:46:20 -04:00 committed by GitHub
parent 075dddf3d3
commit 8db03f3ab2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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}'