File tree Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -104,6 +104,10 @@ if(WITH_GPU)
104
104
if (${CUDNN_MAJOR_VERSION} VERSION_LESS 7 )
105
105
message (FATAL_ERROR "Anakin needs CUDNN >= 7.0 to compile" )
106
106
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} " )
107
111
endif ()
108
112
elseif (WITH_AMD_GPU )
109
113
add_definitions (-DPADDLE_WITH_HIP )
Original file line number Diff line number Diff line change @@ -37,7 +37,7 @@ ExternalProject_Add(
37
37
${EXTERNAL_PROJECT_LOG_ARGS}
38
38
# TODO(luotao): use PaddlePaddle/Anakin later
39
39
GIT_REPOSITORY "https://github.com/luotao1/Anakin"
40
- GIT_TAG "3957ae9263eaa0b1986758dac60a88852afb09be "
40
+ GIT_TAG "842a89ae3747ede25d8acbc29030d2eb602ced1f "
41
41
PREFIX ${ANAKIN_SOURCE_DIR}
42
42
UPDATE_COMMAND ""
43
43
CMAKE_ARGS -DUSE_GPU_PLACE=YES
You can’t perform that action at this time.
0 commit comments