diff --git a/docker/Dockerfile b/docker/Dockerfile
index ff08f5e..f16c5d6 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -112,7 +112,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     hipblas-dev hiprand-dev hipsparse-dev hipsparselt-dev \
     hipsolver-dev hipcub-dev rocprim-dev rocthrust-dev \
     rocblas-dev rocrand-dev hipfft-dev hipblaslt \
-    rocm-cmake && \
+    rocm-cmake rocm-core-dev && \
     apt-get clean && rm -rf /var/lib/apt/lists/*
 
 # ===========================================================================