diff --git a/docker/Dockerfile b/docker/Dockerfile index 513480a..1dcfa12 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -128,7 +128,8 @@ RUN python${PYTHON_VER} -m venv ${VENV} && \ 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 && \ + rocblas-dev rocrand-dev hipfft-dev hipblaslt \ + rocm-cmake && \ apt-get clean && rm -rf /var/lib/apt/lists/* # ===========================================================================