frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Correction to paths to find cuDNN

Browse Source
This commit is contained in:
Simon Layton
2016-12-16 16:03:23 -05:00
parent dac78727fb
commit 99e97a4b7a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmake/Modules/FindCuDNN.cmake
View File

@ -19,7 +19,7 @@ find_path(CUDNN_INCLUDE_DIR cudnn.h
find_library(CUDNN_LIBRARY cudnn
PATHS ${CUDNN_ROOT_DIR}
PATH_SUFFIXES lib cuda/lib)
PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64)
find_package_handle_standard_args(CUDNN DEFAULT_MSG CUDNN_INCLUDE_DIR CUDNN_LIBRARY)