diff --git a/docker-compose.amdgpu.yaml b/docker-compose.amdgpu.yaml
index 130b1ccb7..7a1295d94 100644
--- a/docker-compose.amdgpu.yaml
+++ b/docker-compose.amdgpu.yaml
@@ -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}'
\ No newline at end of file
+      - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION-11.0.0}'
\ No newline at end of file