aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/docker/Dockerfile.gpu
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.gpu')
-rw-r--r--tensorflow/tools/docker/Dockerfile.gpu7
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu
index 8c88c323e2..ca3252e1d9 100644
--- a/tensorflow/tools/docker/Dockerfile.gpu
+++ b/tensorflow/tools/docker/Dockerfile.gpu
@@ -30,11 +30,10 @@ RUN pip --no-cache-dir install \
numpy \
scipy \
sklearn \
+ Pillow \
&& \
python -m ipykernel.kernelspec
-ENV TENSORFLOW_VERSION 0.11.0
-
# --- DO NOT EDIT OR DELETE BETWEEN THE LINES --- #
# These lines will be edited automatically by parameterized_docker_build.sh. #
# COPY _PIP_FILE_ /
@@ -43,9 +42,11 @@ ENV TENSORFLOW_VERSION 0.11.0
# Install TensorFlow GPU version.
RUN pip --no-cache-dir install \
- http://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-${TENSORFLOW_VERSION}-cp27-none-linux_x86_64.whl
+ http://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.0.0-cp27-none-linux_x86_64.whl
# --- ~ DO NOT EDIT OR DELETE BETWEEN THE LINES --- #
+# RUN ln -s /usr/bin/python3 /usr/bin/python#
+
# Set up our notebook config.
COPY jupyter_notebook_config.py /root/.jupyter/