From 90ed6f5d79bc127ae9d51c2298671281884eab0b Mon Sep 17 00:00:00 2001 From: Tinson Lai Date: Wed, 14 Dec 2022 15:36:45 +0800 Subject: [PATCH] Add custom CUDNN finding paths for 64bit Windows (#49066) --- cmake/cudnn.cmake | 2 ++ 1 file changed, 2 insertions(+) diff --git a/cmake/cudnn.cmake b/cmake/cudnn.cmake index 34cc9c8199b..04da3db9519 100644 --- a/cmake/cudnn.cmake +++ b/cmake/cudnn.cmake @@ -29,11 +29,13 @@ list( ${CUDNN_ROOT} ${CUDNN_ROOT}/lib64 ${CUDNN_ROOT}/lib + ${CUDNN_ROOT}/lib/x64 ${CUDNN_ROOT}/lib/${TARGET_ARCH}-linux-gnu ${CUDNN_ROOT}/local/cuda-${CUDA_VERSION}/targets/${TARGET_ARCH}-linux/lib/ $ENV{CUDNN_ROOT} $ENV{CUDNN_ROOT}/lib64 $ENV{CUDNN_ROOT}/lib + $ENV{CUDNN_ROOT}/lib/x64 /usr/lib ${CUDA_TOOLKIT_ROOT_DIR} ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64) -- GitLab