diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm
index bd7cc4a242fa..c4383a0eed5d 100644
--- a/docker/Dockerfile.rocm
+++ b/docker/Dockerfile.rocm
@@ -124,7 +124,8 @@ RUN pip install psutil \
                 numpy \
                 sklearn \
                 scikit-learn \
-                mpi4py 
+                mpi4py \
+                h5py 
 
 ##############################################################################
 ## SSH daemon port inside container cannot conflict with host OS port