Fix anakin build problem in manylinux1 docker image

7 years ago · e10e0d4a5b
parent e525aa232e
commit e10e0d4a5b
1 changed files with 3 additions and 1 deletions
--- 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)