first commit
This commit is contained in:
@@ -0,0 +1,23 @@
|
||||
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 -id --cmake_install
|
||||
WORKDIR /root
|
||||
|
||||
## need to fix broken version
|
||||
CMD [ "/bin/bash" ]
|
||||
Reference in New Issue
Block a user