diff options
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.devel-gpu')
| -rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel-gpu | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu index 56de5940ab..3c85d16a9d 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:7.0-cudnn2-devel +FROM nvidia/cuda:7.5-cudnn4-devel MAINTAINER Craig Citro <craigcitro@google.com> @@ -64,7 +64,7 @@ RUN echo "build --spawn_strategy=standalone --genrule_strategy=standalone" \ >>/root/.bazelrc ENV BAZELRC /root/.bazelrc # Install the most recent bazel release. -ENV BAZEL_VERSION 0.1.4 +ENV BAZEL_VERSION 0.2.0 WORKDIR / RUN mkdir /bazel && \ cd /bazel && \ @@ -96,7 +96,6 @@ WORKDIR /root # Set up CUDA variables ENV CUDA_PATH /usr/local/cuda -ENV LD_LIBRARY_PATH /usr/local/cuda/lib64 # TensorBoard EXPOSE 6006 |