From b0790421d302500472620badaafee68474687d24 Mon Sep 17 00:00:00 2001 From: Per Goncalves da Silva Date: Wed, 15 May 2019 11:28:41 +0200 Subject: [PATCH] Sets cudnn to version 7.5.0 in tensorrt environment --- ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt | 6 +++++- ci/docker/install/ubuntu_cudnn.sh | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt b/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt index 8ad90aedeb6f..09591ee0b9c9 100644 --- a/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt +++ b/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt @@ -18,7 +18,7 @@ # # Dockerfile to run MXNet on Ubuntu 16.04 for CPU -FROM nvidia/cuda:10.0-cudnn7-devel +FROM nvidia/cuda:10.0-devel WORKDIR /work/deps @@ -35,6 +35,10 @@ ARG USER_ID=0 COPY install/ubuntu_adduser.sh /work/ RUN /work/ubuntu_adduser.sh +ENV CUDNN_VERSION=7.5.0.56 +COPY install/ubuntu_cudnn.sh /work/ +RUN /work/ubuntu_cudnn.sh + COPY runtime_functions.sh /work/ WORKDIR /work/mxnet diff --git a/ci/docker/install/ubuntu_cudnn.sh b/ci/docker/install/ubuntu_cudnn.sh index 3f649074f4d5..3d260046b5e7 100755 --- a/ci/docker/install/ubuntu_cudnn.sh +++ b/ci/docker/install/ubuntu_cudnn.sh @@ -50,5 +50,5 @@ case ${CUDA_VERSION} in ;; esac -apt-get install -y libcudnn7=${libcudnn7_version} libcudnn7-dev=${libcudnn7_dev_version} +apt-get install -y --allow-downgrades libcudnn7=${libcudnn7_version} libcudnn7-dev=${libcudnn7_dev_version}