NVIDIA/cutlass
1
0
Fork 0
mirror of https://github.com/NVIDIA/cutlass.git synced 2026-04-20 06:48:59 +00:00
Code Issues 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(
_CUDNN_INCLUDE_DIR cudnn.h
PATHS
$ENV{CUDNN_PATH}
${CUDA_TOOLKIT_ROOT_DIR}/include
$ENV{CUDNN_PATH}/include
$ENV{CUDA_PATH}/include
@@ -48,6 +49,7 @@ find_path(
find_library(
_CUDNN_LIBRARY cudnn
HINTS
$ENV{CUDNN_PATH}
${CUDA_TOOLKIT_ROOT_DIR}/lib64
${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib