nvidia-docker上でTensorFlowを実行するDockerfile

Dockerfile.tensorflow

FROM nvidia/cuda:7.5-cudnn5-devel-centos7

MAINTAINER Takashi Kitano <@kashitan>

RUN yum install -y epel-release && \
    yum install -y gcc \
                   gcc-c++ \
                   initscripts \
                   libcurl-devel \
                   make \
                   openssl-devel \
                   python34-devel && \
    yum clean all

# Install pip
RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
    python3 get-pip.py && \
    rm get-pip.py

# Install TensorFlow
RUN pip --no-cache-dir install \
        ipykernel \
        jupyter \
        matplotlib \
        numpy \
        scipy \
        && \
    python3 -m ipykernel.kernelspec

ENV TF_BINARY_URL https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp34-cp34m-linux_x86_64.whl
RUN pip install --no-cache-dir --upgrade $TF_BINARY_URL

# Set up our notebook config.
COPY jupyter_notebook_config.py /root/.jupyter/

# We just add a little wrapper script.
COPY run_jupyter.sh /
RUN chmod a+x /run_jupyter.sh

# TensorBoard
EXPOSE 6006

# IPython
EXPOSE 8888

WORKDIR "/notebooks"
CMD ["/run_jupyter.sh"]