Skip to content

Commit f6f5552

Browse files
authored
fix CUDNN_VERSION (#76145)
* fix CUDNN_VERSION * fix * fix
1 parent 1cccb88 commit f6f5552

File tree

1 file changed

+6
-18
lines changed

1 file changed

+6
-18
lines changed

cmake/configure.cmake

Lines changed: 6 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -106,16 +106,16 @@ if(WITH_GPU)
106106

107107
find_package(CUDA REQUIRED)
108108

109-
if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_LESS 10.1)
110-
message(FATAL_ERROR "Paddle needs CUDA >= 10.1 to compile")
109+
if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_LESS 11.7)
110+
message(FATAL_ERROR "Paddle needs CUDA >= 11.7 to compile")
111111
endif()
112112

113113
if(NOT CUDNN_FOUND)
114114
message(FATAL_ERROR "Paddle needs cudnn to compile")
115115
endif()
116116

117-
if(${CUDNN_MAJOR_VERSION} VERSION_LESS 7)
118-
message(FATAL_ERROR "Paddle needs CUDNN >= 7.0 to compile")
117+
if(${CUDNN_MAJOR_VERSION} VERSION_LESS 8)
118+
message(FATAL_ERROR "Paddle needs CUDNN >= 8.0 to compile")
119119
endif()
120120

121121
if(CUPTI_FOUND)
@@ -131,20 +131,8 @@ if(WITH_GPU)
131131
include_directories(${CUDA_TOOLKIT_INCLUDE})
132132

133133
if(TENSORRT_FOUND)
134-
if(WIN32)
135-
if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_LESS 9)
136-
message(FATAL_ERROR "TensorRT needs CUDA >= 9.0 to compile on Windows")
137-
endif()
138-
else()
139-
if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_LESS 8)
140-
message(FATAL_ERROR "TensorRT needs CUDA >= 8.0 to compile")
141-
endif()
142-
if(${CUDNN_MAJOR_VERSION} VERSION_LESS 7)
143-
message(FATAL_ERROR "TensorRT needs CUDNN >= 7.0 to compile")
144-
endif()
145-
if(${TENSORRT_MAJOR_VERSION} VERSION_LESS 4)
146-
message(FATAL_ERROR "Paddle needs TensorRT >= 4.0 to compile")
147-
endif()
134+
if(${TENSORRT_MAJOR_VERSION} VERSION_LESS 8)
135+
message(FATAL_ERROR "Paddle needs TensorRT >= 8.0 to compile")
148136
endif()
149137
include_directories(${TENSORRT_INCLUDE_DIR})
150138
endif()

0 commit comments

Comments
 (0)