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

access2rohit · Mar 10, 2021 · 7efc833 · 7efc833
1 parent 52c4e5a
commit 7efc833
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"
 		;;