Skip to content

Commit 3373535

Browse files
committed
fix specific cudnn include and library path
1 parent 83c85f3 commit 3373535

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

cmake/configure.cmake

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,10 @@ if(WITH_GPU)
104104
if(${CUDNN_MAJOR_VERSION} VERSION_LESS 7)
105105
message(FATAL_ERROR "Anakin needs CUDNN >= 7.0 to compile")
106106
endif()
107+
set(ENV{CUDNN_INCLUDE_DIR} ${CUDNN_INCLUDE_DIR})
108+
set(ENV{CUDNN_LIBRARY} ${CUDNN_LIBRARY})
109+
message(STATUS "cudnn include header is ${CUDNN_INCLUDE_DIR}/cudnn.h")
110+
message(STATUS "cudnn library is ${CUDNN_LIBRARY}")
107111
endif()
108112
elseif(WITH_AMD_GPU)
109113
add_definitions(-DPADDLE_WITH_HIP)

cmake/external/anakin.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ ExternalProject_Add(
3737
${EXTERNAL_PROJECT_LOG_ARGS}
3838
# TODO(luotao): use PaddlePaddle/Anakin later
3939
GIT_REPOSITORY "https://github.com/luotao1/Anakin"
40-
GIT_TAG "3957ae9263eaa0b1986758dac60a88852afb09be"
40+
GIT_TAG "842a89ae3747ede25d8acbc29030d2eb602ced1f"
4141
PREFIX ${ANAKIN_SOURCE_DIR}
4242
UPDATE_COMMAND ""
4343
CMAKE_ARGS -DUSE_GPU_PLACE=YES

0 commit comments

Comments
 (0)