Add custom CUDNN finding paths for 64bit Windows (PaddlePaddle#49066)

lxsbupt · Dec 14, 2022 · 03fbd81 · 03fbd81
1 parent 3b12162
commit 03fbd81
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)