tensorflow docker 直接使用会导致一些不方便,最好根据需求安装一些python包,写下来为了以后使用:
FROM tensorflow/tensorflow:2.0.0rc0-gpu-py3
RUN mkdir ~/.pip
# pip源
RUN echo -e "[global]\n"\
"index-url = http://mirrors.aliyun.com/pypi/simple/\n"\
"[install]\n"\
"trusted-host=mirrors.aliyun.com" > ~/.pip/pip.conf
RUN python -m pip install pillow scipy opencv-python
# 安装vim
RUN apt install -y vim \
libsm6 \
libxrender1 \
libxext6