@@ -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