|
|
|
@ -104,6 +104,10 @@ if(WITH_GPU)
|
|
|
|
|
if(${CUDNN_MAJOR_VERSION} VERSION_LESS 7)
|
|
|
|
|
message(FATAL_ERROR "Anakin needs CUDNN >= 7.0 to compile")
|
|
|
|
|
endif()
|
|
|
|
|
set(ENV{CUDNN_INCLUDE_DIR} ${CUDNN_INCLUDE_DIR})
|
|
|
|
|
set(ENV{CUDNN_LIBRARY} ${CUDNN_LIBRARY})
|
|
|
|
|
message(STATUS "cudnn include header is ${CUDNN_INCLUDE_DIR}/cudnn.h")
|
|
|
|
|
message(STATUS "cudnn library is ${CUDNN_LIBRARY}")
|
|
|
|
|
endif()
|
|
|
|
|
elseif(WITH_AMD_GPU)
|
|
|
|
|
add_definitions(-DPADDLE_WITH_HIP)
|
|
|
|
|