diff options
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.gpu')
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.gpu | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu index 77699ebb42..8d2ede62b5 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -1,9 +1,10 @@ -FROM nvidia/cuda:7.0-cudnn2-runtime +FROM nvidia/cuda:7.5-cudnn4-runtime MAINTAINER Craig Citro <craigcitro@google.com> # Pick up some TF dependencies RUN apt-get update && apt-get install -y \ + bc \ curl \ libfreetype6-dev \ libpng12-dev \ @@ -28,13 +29,16 @@ RUN pip --no-cache-dir install \ python -m ipykernel.kernelspec # Install TensorFlow GPU version. -ENV TENSORFLOW_VERSION 0.7.0 +ENV TENSORFLOW_VERSION 0.7.1 RUN pip --no-cache-dir install \ - http://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-${TENSORFLOW_VERSION}-py2-none-linux_x86_64.whl + http://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-${TENSORFLOW_VERSION}-cp27-none-linux_x86_64.whl # Set up our notebook config. COPY jupyter_notebook_config.py /root/.jupyter/ +# Copy sample notebooks. +COPY notebooks /notebooks + # Jupyter has issues with being run directly: # https://github.com/ipython/ipython/issues/7062 # We just add a little wrapper script. @@ -45,6 +49,6 @@ EXPOSE 6006 # IPython EXPOSE 8888 -WORKDIR "/root" +WORKDIR "/notebooks" -CMD ["/bin/bash"] +CMD ["/run_jupyter.sh"] |