FROM ubuntu:18.04

ENV LANG C.UTF-8

ENV LANGUAGE C.UTF-8

ENV LC_ALL C.UTF-8

RUN sed -i s@/archive.ubuntu.com/@/mirrors.tencent.com/@g /etc/apt/sources.list

RUN sed -i s@/security.ubuntu.com/@/mirrors.tencent.com/@g /etc/apt/sources.list

RUN apt-get clean && apt-get update -y && apt-get -y install --no-install-recommends apt-utils

RUN apt-get -y install git cmake make

RUN apt-get -y install gcc g++

RUN apt-get -y install protobuf-compiler libprotobuf-dev

RUN apt-get -y install python3 python3-dev python3-pip

#RUN mkdir -p /root/.pip && echo "[global]\n index-url = https://mirrors.tencent.com/pypi/simple/" >> /root/.pip/pip.conf

RUN python3 -m pip install --upgrade pip && pip3 install -U onnxruntime numpy onnx-simplifier setuptools protobuf

RUN pip3 install tensorflow==1.15.0 tf2onnx

ENV TNN_ROOT=/opt/TNN

ENV TOOLS_ROOT=$TNN_ROOT/tools

# COPY ./onnx2tnn $TOOLS_ROOT/onnx2tnn

# COPY ./caffe2onnx $TOOLS_ROOT/caffe2onnx

# COPY ./convert2tnn $TOOLS_ROOT/convert2tnn

COPY . $TNN_ROOT/

#RUN cd $TOOLS_ROOT/onnx2tnn/onnx-converter && ./build.sh

RUN cd $TOOLS_ROOT/convert2tnn && bash ./build.sh

RUN python3 $TOOLS_ROOT/convert2tnn/converter.py -h

WORKDIR $TOOLS_ROOT/convert2tnn/