Set the default cuDNN installation path

1eb98e2f · Yi Wang · ab270c38 · 1eb98e2f
隐藏空白更改
内联并排

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}