FROM ubuntu_arm:18.04

ARG NNAE_VERSION
ARG NNAR_ARCH=arm64
ARG HOST_ASCEND_BASE=/usr/local/Ascend
ARG CNT_ASCEND_BASE=/usr/local/Ascend/nnae/$NNAE_VERSION/$NNAR_ARCH-linux_gcc7.3.0

ARG INSTALL_ASCEND_PAGS_SH=install_ascend_pkgs.sh
WORKDIR /var/tmp
COPY . ./
RUN bash $INSTALL_ASCEND_PAGS_SH $NNAE_VERSION
RUN bash -c "rm -rf /var/tmp/*"

WORKDIR /root
ENV LD_LIBRARY_PATH=\
$HOST_ASCEND_BASE/add-ons:\
$CNT_ASCEND_BASE/fwkacllib/lib64:\
$HOST_ASCEND_BASE/driver/lib64/driver:\
$HOST_ASCEND_BASE/driver/lib64/common:$LD_LIBRARY_PATH

ENV GLOG_v=2
ENV TBE_IMPL_PATH=${CNT_ASCEND_BASE}/opp/op_impl/built-in/ai_core/tbe/
ENV PYTHONPATH=$TBE_IMPL_PATH:$PYTHONPATH
ENV PATH=${CNT_ASCEND_BASE}/fwkacllib/ccec_compiler/bin/:${PATH}