说明
- ubuntu换源;
- 构建pth->onnx->trt转模型需要的环境;
dockerfile代码
FROM nvcr.io/nvidia/tensorrt:20.02-py3
LABEL maintainer="XINWEN"
RUN mv /etc/apt/sources.list /etc/apt/sources.list.bak
RUN echo "deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb-src http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb-src http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse" >> /etc/apt/sources.list \
&& echo "deb-src http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse" >> /etc/apt/sources.list \
&& apt-get update \
&& apt-get -y install clang-format nfs-common \
&& apt-get -y install openssh-client
RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && python3 get-pip.py --force-reinstall
ADD ./soft/* /softs/
WORKDIR /softs
RUN pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip \
&& pip3 install torch-1.3.0-cp36-cp36m-manylinux1_x86_64.whl \
&& pip3 install Pillow-6.0.0-cp36-cp36m-manylinux1_x86_64.whl \
&& pip3 install h5py-2.10.0-cp36-cp36m-manylinux1_x86_64.whl \
&& pip3 install grpcio-1.27.2-cp36-cp36m-manylinux2010_x86_64.whl \
&& pip3 install tensorboard-1.15.0-py3-none-any.whl \
&& pip3 install onnx-1.6.0-cp36-cp36m-manylinux1_x86_64.whl \
&& pip3 install opencv_python-4.2.0.32-cp36-cp36m-manylinux1_x86_64.whl \
&& pip3 install tensorflow_gpu-1.15.0-cp36-cp36m-manylinux2010_x86_64.whl \
&& pip3 install torchvision-0.4.1-cp36-cp36m-manylinux1_x86_64.whl \
&& pip3 install onnxruntime-1.1.0-cp36-cp36m-manylinux1_x86_64.whl \
&& cd onnx-simplifier-0.2.2 && python3 setup.py install \
# RUN pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip \
# && pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.3.0 torchvision==0.4.1 \
# && pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple onnx==1.6.0 onnxruntime==1.1.0 onnx-simplifier==0.2.2 \
# && pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv_python==4.2.0 Pillow==6.0.0 \
# && pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow_gpu==1.15.0 \
&& pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple pytest pyhamcrest pytest-cov black==19.3b0 isort==4.3.21 flake8 'pillow<7.0.0' cython dvc
WORKDIR /workspace
RUN rm -rf /softs
# RUN apt-get -y install isort clang-format nfs-common
ENTRYPOINT /bin/bash