diff --git a/cmake/Cuda.cmake b/cmake/Cuda.cmake index e03feabffcb..681c8c375ca 100644 --- a/cmake/Cuda.cmake +++ b/cmake/Cuda.cmake @@ -198,7 +198,11 @@ function(detect_cuDNN) set(HAVE_CUDNN TRUE PARENT_SCOPE) set(CUDNN_FOUND TRUE PARENT_SCOPE) - file(READ ${CUDNN_INCLUDE}/cudnn.h CUDNN_VERSION_FILE_CONTENTS) + if(EXISTS ${CUDNN_INCLUDE}/cudnn_version.h) + file(READ ${CUDNN_INCLUDE}/cudnn_version.h CUDNN_VERSION_FILE_CONTENTS) + else() + file(READ ${CUDNN_INCLUDE}/cudnn.h CUDNN_VERSION_FILE_CONTENTS) + endif() # cuDNN v3 and beyond string(REGEX MATCH "define CUDNN_MAJOR * +([0-9]+)"