tensorflow docker 直接使用會(huì)導(dǎo)致一些不方便,最好根據(jù)需求安裝一些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