diff --git a/docker/amd-docker.Dockerfile b/docker/amd-docker.Dockerfile
index 7d2d3ae5..64d4316d 100644
--- a/docker/amd-docker.Dockerfile
+++ b/docker/amd-docker.Dockerfile
@@ -57,3 +57,5 @@ RUN sudo pip install \
     packaging \
     wheel \
     tinygrad
+
+RUN sudo pip install git+https://github.com/ROCm/iris.git