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

90ed6f5d · Tinson Lai · GitHub · fd3169da · 90ed6f5d
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

cmake/cudnn.cmake cmake/cudnn.cmake +2 -0

未找到文件。
--- 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)