From 03fbd81aab75d8fef667b0940c23d2e72e7f603d 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 34cc9c8199b37..04da3db951972 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)