Add cudnn v5 path in CMAKE

4aa41233 · liaogang · 699da5be · 4aa41233
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

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

未找到文件。
--- a/cmake/cudnn.cmake
+++ b/cmake/cudnn.cmake
@@ -11,6 +11,7 @@ list(APPEND CUDNN_CHECK_LIBRARY_DIRS
    ${CUDNN_ROOT}
    ${CUDNN_ROOT}/lib64
    ${CUDNN_ROOT}/lib
+    ${CUDNN_ROOT}/lib/x86_64-linux-gnu
    $ENV{CUDNN_ROOT}
    $ENV{CUDNN_ROOT}/lib64
    $ENV{CUDNN_ROOT}/lib