From eb51b8b8c629ad4ce2abeceecafe3686b8661273 Mon Sep 17 00:00:00 2001 From: stu1130 Date: Thu, 5 Dec 2019 13:14:30 -0800 Subject: [PATCH] Downgrade the cublas version for nightly gpu --- ci/docker/Dockerfile.build.ubuntu_base_gpu | 4 ++++ ci/docker/Dockerfile.build.ubuntu_nightly_gpu | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/ci/docker/Dockerfile.build.ubuntu_base_gpu b/ci/docker/Dockerfile.build.ubuntu_base_gpu index 94e6437e578b..daf6fa342d43 100644 --- a/ci/docker/Dockerfile.build.ubuntu_base_gpu +++ b/ci/docker/Dockerfile.build.ubuntu_base_gpu @@ -29,6 +29,10 @@ ENV CUDNN_VERSION=7.6.0.64 COPY install/ubuntu_cudnn.sh /work/ RUN /work/ubuntu_cudnn.sh +# hotfix nvidia-docker image come with wrong version of libcublas +COPY install/ubuntu_cublas.sh /work/ +RUN /work/ubuntu_cublas.sh + ARG USER_ID=0 ARG GROUP_ID=0 COPY install/ubuntu_adduser.sh /work/ diff --git a/ci/docker/Dockerfile.build.ubuntu_nightly_gpu b/ci/docker/Dockerfile.build.ubuntu_nightly_gpu index 5e812c433b43..9a6781685031 100644 --- a/ci/docker/Dockerfile.build.ubuntu_nightly_gpu +++ b/ci/docker/Dockerfile.build.ubuntu_nightly_gpu @@ -72,6 +72,10 @@ ENV CUDNN_VERSION=7.6.0.64 COPY install/ubuntu_cudnn.sh /work/ RUN /work/ubuntu_cudnn.sh +# hotfix nvidia-docker image come with wrong version of libcublas +COPY install/ubuntu_cublas.sh /work/ +RUN /work/ubuntu_cublas.sh + ARG USER_ID=0 ARG GROUP_ID=0 COPY install/ubuntu_adduser.sh /work/