FROM ubuntu_arm:18.04 AS tmp
ADD Ascend-cann-toolkit*.run /root
ADD whl /root
ADD version.info /root
ADD ascend_install.info /root
WORKDIR /root
RUN useradd HwHiAiUser
RUN cp ascend_install.info /etc/
RUN mkdir -p /usr/local/Ascend/driver
RUN cp version.info /usr/local/Ascend/driver/
RUN bash Ascend-cann-toolkit*.run --install
RUN rm -rf /usr/local/Ascend/driver
RUN pip3 install mindspore_ascend*.whl
RUN pip3 install mindspore_serving*.whl
RUN pip3 install /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/atc/lib64/te*.whl
RUN pip3 install /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/atc/lib64/topi*.whl
RUN rm -rf /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/fwkacllib
RUN rm -rf /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/toolkit
RUN rm -rf /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/atc/python/site-packages/*
RUN rm -rf /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/atc/lib64/*.whl
RUN rm -rf /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/atc/lib64/*.a
RUN rm -rf /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/acllib/lib64/*.a
RUN mv /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/opp/data/tiling/ascend310 /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/opp/data/a310_bak && rm -rf /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/opp/data/tiling && mkdir -p /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/opp/data/tiling && mv /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/opp/data/a310_bak /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/opp/data/tiling/ascend310

FROM arm64v8/ubuntu:18.04
COPY --from=tmp /usr/local/Ascend /usr/local/Ascend
COPY --from=tmp /usr/local/python /usr/local/python
COPY --from=tmp /usr/lib/aarch64-linux-gnu/libstdc++.so.6.0.25 /usr/lib/aarch64-linux-gnu
RUN cd /usr/lib/aarch64-linux-gnu/ && rm libstdc++.so.6 && ln -s libstdc++.so.6.0.25 libstdc++.so.6  \
    && sed -i "s@http://ports.ubuntu.com@http://mirrors.huaweicloud.com@g" /etc/apt/sources.list \
    && apt update \
    && apt install -y sudo libnuma-dev libffi-dev libssl-dev zlib1g-dev libbz2-dev libncurses5-dev \
       libgdbm-dev liblzma-dev libreadline-dev libsm6 libjpeg-dev libfreetype6-dev libblas-dev \
       liblapack-dev libopenblas-dev libatlas-base-dev libxml2 \
    && ln -s /lib /lib64
ENV PATH /usr/local/python/python375/bin:/usr/local/Ascend/ascend-toolkit/latest/arm64-linux/atc/bin:/usr/local/Ascend/ascend-toolkit/latest/arm64-linux/atc/ccec_compiler/bin:$PATH
ENV LD_LIBRARY_PATH /usr/local/python/python375/lib:/usr/local/python/python375/lib/python3.7/site-packages/mindspore/lib:/usr/local/Ascend/driver/lib64:/usr/local/Ascend/ascend-toolkit/latest/arm64-linux/atc/lib64:/usr/local/Ascend/ascend-toolkit/latest/arm64-linux/acllib/lib64$LD_LIBRARY_PATH
ENV TBE_IMPL_PATH /usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe
ENV ASCEND_OPP_PATH /usr/local/Ascend/ascend-toolkit/latest/opp