Merge pull request #3581 from wangkuiyi/default_value_for_cudnn_root

Set the default cuDNN installation path
8 years ago · 0d9846f3c0
parent ac663353e2 1eb98e2fef
commit 0d9846f3c0
1 changed files with 1 additions and 1 deletions
--- a/cmake/cudnn.cmake
+++ b/cmake/cudnn.cmake
@ -2,7 +2,7 @@ if(NOT WITH_GPU)
    return()
 endif()

-set(CUDNN_ROOT "" CACHE PATH "CUDNN ROOT")
+set(CUDNN_ROOT "/usr" CACHE PATH "CUDNN ROOT")
 find_path(CUDNN_INCLUDE_DIR cudnn.h
    PATHS ${CUDNN_ROOT} ${CUDNN_ROOT}/include
    $ENV{CUDNN_ROOT} $ENV{CUDNN_ROOT}/include ${CUDA_TOOLKIT_INCLUDE}