24 lines
836 B
Docker
24 lines
836 B
Docker
FROM ubuntu:22.04
|
|
ARG ROCM_VERSION="5.7.1"
|
|
ARG ROCBLAS_BRANCH="release/rocm-rel-5.7"
|
|
ARG DEBIAN_FRONTEND="noninteractive"
|
|
RUN apt-get update \
|
|
&& apt-get install -y wget git \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
WORKDIR /root
|
|
RUN wget -O amdgpu-install.deb https://repo.radeon.com/amdgpu-install/${ROCM_VERSION}/ubuntu/jammy/amdgpu-install_5.7.50701-1_all.deb
|
|
RUN dpkg -i amdgpu-install.deb
|
|
RUN amdgpu-install -y --no-dkms --usecase=rocm
|
|
RUN git clone -b ${ROCBLAS_BRANCH} https://github.com/ROCmSoftwarePlatform/rocBLAS.git
|
|
WORKDIR /root/rocBLAS
|
|
RUN rm -rf library/src/blas3/Tensile/Logic/asm_full/r9nano/
|
|
|
|
# gfx803 as single target will cause error
|
|
RUN sed -i -e 's/rmake.py --install/rmake.py -a="gfx803;gfx900" --install/g' install.sh
|
|
|
|
RUN ./install.sh -d --cmake_install
|
|
WORKDIR /root
|
|
|
|
## need to fix broken version
|
|
CMD [ "/bin/bash" ]
|