Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Downgrade the cublas version for nightly gpu (#16985)
Browse files Browse the repository at this point in the history
  • Loading branch information
@stu1130 @lanking520
stu1130 authored and lanking520 committed Dec 6, 2019
1 parent 10903e8 commit 8dd7051
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 0 deletions.
4 changes: 4 additions & 0 deletions ci/docker/Dockerfile.build.ubuntu_base_gpu
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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/
Expand Down
4 changes: 4 additions & 0 deletions ci/docker/Dockerfile.build.ubuntu_nightly_gpu
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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/
Expand Down

0 comments on commit 8dd7051

Please sign in to comment.