diff options
Diffstat (limited to 'tensorflow/tools/ci_build/Dockerfile.gpu')
-rw-r--r-- | tensorflow/tools/ci_build/Dockerfile.gpu | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu b/tensorflow/tools/ci_build/Dockerfile.gpu index b57d1d18c1..b4b0ccccf7 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu +++ b/tensorflow/tools/ci_build/Dockerfile.gpu @@ -1,13 +1,12 @@ -FROM nvidia/cuda:7.0-cudnn2-devel +FROM nvidia/cuda:7.5-cudnn4-devel MAINTAINER Jan Prach <jendap@google.com> # Copy and run the install scripts. -COPY install/install_deb_packages.sh /install/install_deb_packages.sh +COPY install/*.sh /install/ +RUN /install/install_bootstrap_deb_packages.sh +RUN add-apt-repository -y ppa:openjdk-r/ppa RUN /install/install_deb_packages.sh -COPY install/install_openjdk8_from_ppa.sh /install/install_openjdk8_from_ppa.sh -RUN /install/install_openjdk8_from_ppa.sh -COPY install/install_bazel.sh /install/install_bazel.sh RUN /install/install_bazel.sh # Set up bazelrc. |