Create a gist now

Instantly share code, notes, and snippets.

What would you like to do?
nvidia-docker上でTensorFlowを実行するDockerfile
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"]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment