File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -31,7 +31,7 @@ ENV DEBIAN_FRONTEND=noninteractive
3131
3232# Install Python and other dependencies
3333RUN apt-get update -y \
34- && apt-get install -y software-properties-common git curl sudo vim less \
34+ && apt-get install -y software-properties-common git curl sudo vim less libgfortran5 \
3535 && add-apt-repository ppa:deadsnakes/ppa \
3636 && apt-get update -y \
3737 && apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv \
@@ -60,7 +60,8 @@ RUN cd hipBLAS-common \
6060RUN git clone https://github.com/ROCm/hipBLASLt
6161RUN cd hipBLASLt \
6262 && git checkout ${HIPBLASLT_BRANCH} \
63- && ./install.sh -d --architecture ${PYTORCH_ROCM_ARCH} ${LEGACY_HIPBLASLT_OPTION} \
63+ && apt-get install -y llvm-dev \
64+ && ./install.sh -dc --architecture ${PYTORCH_ROCM_ARCH} ${LEGACY_HIPBLASLT_OPTION} \
6465 && cd build/release \
6566 && make package
6667RUN mkdir -p /app/install && cp /app/hipBLASLt/build/release/*.deb /app/hipBLAS-common/build/*.deb /app/install
You can’t perform that action at this time.
0 commit comments