From 2f246ef177ad3cb8aba3b4de0a628822cbeab71d Mon Sep 17 00:00:00 2001 From: Gunhan Gulsoy Date: Fri, 4 Jan 2019 14:27:22 -0800 Subject: [PATCH] Upgrade ci docker images to cuda 10. These really should be gone already... PiperOrigin-RevId: 227914258 --- tensorflow/tools/ci_build/Dockerfile.gpu | 4 ++-- tensorflow/tools/ci_build/Dockerfile.rbe.gpu | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu b/tensorflow/tools/ci_build/Dockerfile.gpu index a4cad4b6c65..ff1fefe8921 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu +++ b/tensorflow/tools/ci_build/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04 +FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 LABEL maintainer="Jan Prach " @@ -24,7 +24,7 @@ COPY install/.bazelrc /etc/bazel.bazelrc ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH # Link NCCL libray and header where the build script expects them. -RUN mkdir /usr/local/cuda-9.0/lib && \ +RUN mkdir /usr/local/cuda/lib && \ ln -s /usr/lib/x86_64-linux-gnu/libnccl.so.2 /usr/local/cuda/lib/libnccl.so.2 && \ ln -s /usr/include/nccl.h /usr/local/cuda/include/nccl.h diff --git a/tensorflow/tools/ci_build/Dockerfile.rbe.gpu b/tensorflow/tools/ci_build/Dockerfile.rbe.gpu index b6562058367..c4912a65b65 100644 --- a/tensorflow/tools/ci_build/Dockerfile.rbe.gpu +++ b/tensorflow/tools/ci_build/Dockerfile.rbe.gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04 +FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 LABEL maintainer="Nick Lopez "