diff options
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.devel-gpu')
| -rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel-gpu | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu index 24350c507e..d0a038a9db 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04 +FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 MAINTAINER Craig Citro <craigcitro@google.com> @@ -57,7 +57,7 @@ RUN echo "startup --batch" >>/etc/bazel.bazelrc RUN echo "build --spawn_strategy=standalone --genrule_strategy=standalone" \ >>/etc/bazel.bazelrc # Install the most recent bazel release. -ENV BAZEL_VERSION 0.4.5 +ENV BAZEL_VERSION 0.5.0 WORKDIR / RUN mkdir /bazel && \ cd /bazel && \ @@ -72,7 +72,7 @@ RUN mkdir /bazel && \ RUN git clone https://github.com/tensorflow/tensorflow.git && \ cd tensorflow && \ - git checkout r1.1 + git checkout r1.2 WORKDIR /tensorflow # Configure the build for our CUDA configuration. |