From d8291858d502f72b4204e362321e1b83012eddf4 Mon Sep 17 00:00:00 2001 From: Stefano Sinigardi Date: Fri, 22 Feb 2019 09:14:41 +0100 Subject: [PATCH] [FindCUDNN] add another hint for newly suggested ENV variable --- cmake/Modules/FindCUDNN.cmake | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cmake/Modules/FindCUDNN.cmake b/cmake/Modules/FindCUDNN.cmake index 75fd5261..ef78833e 100644 --- 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()