From eeaebb648a88ad25b61a684edc30fa8fc2d84ce9 Mon Sep 17 00:00:00 2001
From: kumailf <49893630+kumailf@users.noreply.github.com>
Date: Wed, 22 Mar 2023 07:51:50 +0800
Subject: [PATCH] fix error :Found cuDNN: ver. ??? found

---
 cmake/Cuda.cmake | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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]+)"