From e10e0d4a5b104dd6c0eb9352498f9c20bdb8ec1f Mon Sep 17 00:00:00 2001 From: minqiyang Date: Mon, 20 Aug 2018 15:13:27 +0800 Subject: [PATCH] Fix anakin build problem in manylinux1 docker image --- cmake/configure.cmake | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cmake/configure.cmake b/cmake/configure.cmake index 53454d79f72..005cf1c9327 100644 --- a/cmake/configure.cmake +++ b/cmake/configure.cmake @@ -113,7 +113,9 @@ if(WITH_GPU) endif() if(WITH_ANAKIN) set(ENV{CUDNN_INCLUDE_DIR} ${CUDNN_INCLUDE_DIR}) - set(ENV{CUDNN_LIBRARY} ${CUDNN_LIBRARY}) + get_filename_component(CUDNN_LIBRARY_DIR ${CUDNN_LIBRARY} DIRECTORY) + set(ENV{CUDNN_LIBRARY} ${CUDNN_LIBRARY_DIR}) + message(STATUS "Set Anakin CUDNN_LIBRARY_DIR:" ${CUDNN_LIBRARY_DIR}) endif() elseif(WITH_AMD_GPU) add_definitions(-DPADDLE_WITH_HIP) -- GitLab