2020-09-04 16:24:11 +03:00
|
|
|
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
|
|
|
|
|
# FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu18.04
|
2020-06-24 08:40:41 +03:00
|
|
|
MAINTAINER Wei CUI <weicu@microsoft.com>
|
|
|
|
|
|
|
|
|
|
ENV DEBIAN_FRONTEND noninteractive
|
|
|
|
|
ENV PYTHONDONTWRITEBYTECODE 1
|
|
|
|
|
ENV HIP_PLATFORM hcc
|
|
|
|
|
ENV PATH $PATH:/opt/rocm/bin:/usr/local/nvidia/lib64/bin
|
|
|
|
|
ENV TVM_HOME=/opt/tvm
|
|
|
|
|
ENV HSA_USERPTR_FOR_PAGED_MEM=0
|
|
|
|
|
ENV TF_ROCM_FUSION_ENABLE 1
|
|
|
|
|
|
|
|
|
|
RUN env > /etc/environment
|
|
|
|
|
|
|
|
|
|
RUN apt-get update && apt install -y --no-install-recommends git ca-certificates \
|
|
|
|
|
python3-pip python3-wheel python3-setuptools python3-dev python3-pytest \
|
|
|
|
|
vim less netcat-openbsd inetutils-ping curl patch iproute2 \
|
|
|
|
|
g++ libpci3 libnuma-dev make file openssh-server kmod gdb libopenmpi-dev openmpi-bin \
|
|
|
|
|
autoconf automake autotools-dev libtool \
|
|
|
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
|
|
|
|
RUN curl -sL http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | apt-key add - && \
|
|
|
|
|
printf "deb [arch=amd64] http://repo.radeon.com/rocm/apt/3.7/ xenial main" | tee /etc/apt/sources.list.d/rocm_hip.list && \
|
|
|
|
|
apt update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
|
|
|
|
rocm-dev zlib1g-dev unzip librdmacm-dev rocblas hipsparse rccl rocfft rocrand miopen-hip && apt-get clean && rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
|
|
|
|
RUN /bin/echo -e "set nocindent\nset noautoindent\nset ts=4" > /root/.vimrc
|
|
|
|
|
|
|
|
|
|
RUN pip3 install tornado psutil xgboost==0.80 numpy decorator attrs cmake && rm -rf ~/.cache
|
|
|
|
|
RUN git clone https://github.com/dmlc/tvm $TVM_HOME
|
|
|
|
|
|
|
|
|
|
RUN cd $TVM_HOME && git checkout v0.6 && git submodule init && git submodule update && \
|
|
|
|
|
mkdir -p build && cd build && cp ../cmake/config.cmake . && \
|
|
|
|
|
sed -i 's/LLVM ON/LLVM OFF/g' config.cmake && sed -i 's/CUDA OFF/CUDA ON/g' config.cmake && \
|
|
|
|
|
cmake .. && make -j16
|
|
|
|
|
|
|
|
|
|
ADD engine/tvm_v0.6.patch $TVM_HOME/tvm_v0.6.patch
|
|
|
|
|
RUN cd $TVM_HOME && git apply tvm_v0.6.patch && cd build && make -j16
|
|
|
|
|
|
|
|
|
|
RUN [ -e /usr/lib/x86_64-linux-gnu/libcuda.so.1 ] || ln -s /host/usr/lib/x86_64-linux-gnu/libcuda.so.1 /usr/lib/x86_64-linux-gnu
|
|
|
|
|
RUN ln -sf libcudart.so /usr/local/cuda/targets/x86_64-linux/lib/libcudart_static.a
|
|
|
|
|
|
|
|
|
|
ENV LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:/usr/local/cuda/compat:$LD_LIBRARY_PATH
|
|
|
|
|
ENV PYTHONPATH=/opt/tvm/python:/opt/tvm/topi/python:/opt/tvm/nnvm/python:/antares
|
|
|
|
|
ENV HIP_IGNORE_HCC_VERSION=1
|
|
|
|
|
|
