Add cudnn v5 path in CMAKE

9 years ago · 4aa41233b8
parent 699da5be1a
commit 4aa41233b8
1 changed files with 1 additions and 0 deletions
--- 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