From e7740d93db582b561b2d2415351dd18c53418b88 Mon Sep 17 00:00:00 2001 From: Manu Seth Date: Thu, 12 Nov 2020 12:21:31 +0000 Subject: [PATCH] update cudnn from 7 to 8 for cu102 --- ci/docker/install/ubuntu_cudnn.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ci/docker/install/ubuntu_cudnn.sh b/ci/docker/install/ubuntu_cudnn.sh index 1cc0b722bfc7..0699d80eb86e 100755 --- a/ci/docker/install/ubuntu_cudnn.sh +++ b/ci/docker/install/ubuntu_cudnn.sh @@ -38,7 +38,7 @@ case ${CUDA_VERSION} in export libcudnn_dev_version="${CUDNN_VERSION}-1+cuda11.0" ;; 10\.2*) - export libcudnn_package="libcudnn7" + export libcudnn_package="libcudnn8" export libcudnn_version="${CUDNN_VERSION}-1+cuda10.2" export libcudnn_dev_version="${CUDNN_VERSION}-1+cuda10.2" ;;