diff --git a/ollama/docker-compose.yml b/ollama/docker-compose.yml
index 25ba61f..0522ca5 100755
--- a/ollama/docker-compose.yml
+++ b/ollama/docker-compose.yml
@@ -1,6 +1,6 @@
 services:
   ollama:
-    image: ${DOCKER_REGISTRY:-}ollama/ollama
+    image: ${DOCKER_REGISTRY:-}ollama/ollama:rocm
     container_name: ollama
     restart: ${RESTART:-unless-stopped}
     devices: