Merge pull request #3581 from wangkuiyi/default_value_for_cudnn_root

Set the default cuDNN installation path

Merge pull request #3581 from wangkuiyi/default_value_for_cudnn_root
Set the default cuDNN installation path
0d9846f3 · Yi Wang · GitHub · ac663353 · 1eb98e2f · 0d9846f3
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

cmake/cudnn.cmake cmake/cudnn.cmake +1 -1

未找到文件。
--- 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}