leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix finding cuDNN (#2890)

Browse Source 
* Fix finding cuDNN

* Search $CUDNN_PATH first, not last
This commit is contained in:
TLescoatTFX
2026-03-05 02:51:37 +01:00
committed by GitHub
parent 49e54f2b23
commit a93d86ec83
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cuDNN.cmake
View File

@@ -39,6 +39,7 @@ message(STATUS "Configuring cuDNN ...")
find_path( find_path(
_CUDNN_INCLUDE_DIR cudnn.h _CUDNN_INCLUDE_DIR cudnn.h
PATHS PATHS
$ENV{CUDNN_PATH}
${CUDA_TOOLKIT_ROOT_DIR}/include ${CUDA_TOOLKIT_ROOT_DIR}/include
$ENV{CUDNN_PATH}/include $ENV{CUDNN_PATH}/include
$ENV{CUDA_PATH}/include $ENV{CUDA_PATH}/include
@@ -48,6 +49,7 @@ find_path(
find_library( find_library(
_CUDNN_LIBRARY cudnn _CUDNN_LIBRARY cudnn
HINTS HINTS
$ENV{CUDNN_PATH}
${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib64
${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib ${CUDA_TOOLKIT_ROOT_DIR}/lib