[FindCUDNN] add another hint for newly suggested ENV variable

6 years ago · d8291858d5
parent 20757e217f
commit d8291858d5
1 changed files with 2 additions and 2 deletions
--- a/cmake/Modules/FindCUDNN.cmake
+++ b/cmake/Modules/FindCUDNN.cmake
@ -24,13 +24,13 @@ include(FindPackageHandleStandardArgs)

 if(NOT CUDNN_INCLUDE_DIR)
  find_path(CUDNN_INCLUDE_DIR cudnn.h
-    HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn}
+    HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn} $ENV{CUDNN}
    PATH_SUFFIXES cuda/include include)
 endif()

 if(NOT CUDNN_LIBRARY)
  find_library(CUDNN_LIBRARY cudnn
-    HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn}
+    HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn} $ENV{CUDNN}
    PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64)
 endif()