update cudnn from 7 to 8 for cu102 (#19522)

apache · Nov 12, 2020 · f1ed842 · f1ed842
1 parent 351953c
commit f1ed842
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"
 		;;