tensorflow dockerfile

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

你可能感兴趣的:(tensorflow dockerfile)