From 4aa41233b89db4415dedf8f5debbdfc28fd71e6d Mon Sep 17 00:00:00 2001 From: liaogang Date: Tue, 28 Mar 2017 10:25:32 +0800 Subject: [PATCH] Add cudnn v5 path in CMAKE --- cmake/cudnn.cmake | 1 + 1 file changed, 1 insertion(+) diff --git a/cmake/cudnn.cmake b/cmake/cudnn.cmake index e5b59be1936..5754407d66c 100644 --- 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 -- GitLab