dockerfile:tensorflow:cuda-7.5

代码语言：javascript复制

# mbuckler/tf-faster-rcnn-deps
#
# Dockerfile to hold dependencies for the Tensorflow implmentation
# of Faster RCNN
FROM mbuckler/tensorflow:cuda-7.5
WORKDIR /root
# Get required packages
RUN apt-get update && 
  apt-get install vim 
                  python-pip 
                  python-dev 
                  python-opencv 
                  python-tk 
                  libjpeg-dev 
                  libfreetype6 
                  libfreetype6-dev 
                  zlib1g-dev 
                  cmake 
                  wget 
                  cython 
                  git 
                  -y
                  
# Get required python modules
RUN pip install --upgrade pip
RUN pip install image 
                scipy 
                matplotlib 
                PyYAML 
                numpy 
                easydict 
# Update numpy
RUN pip install -U numpy
# Install python interface for COCO
RUN git clone https://github.com/pdollar/coco.git
WORKDIR /root/coco/PythonAPI
RUN make
WORKDIR /root
# Add CUDA to the path
ENV LD_LIBRARY_PATH $LD_LIBRARY_PATH:/usr/local/cuda/lib64
ENV CUDA_HOME /usr/local/cuda
ENV PYTHONPATH /root/coco/PythonAPI
RUN ldconfig

cuda dockerfile

0 人点赞