3.0.1 Windows cuda128 (#72433)

swgu98 · web-flow · commit f9cb991786a6 · 2025-04-28T10:53:59.000+08:00
* cuda128 cudnn970 trt105018

* fix
diff --git a/cmake/tensorrt.cmake b/cmake/tensorrt.cmake
@@ -4,9 +4,25 @@ endif()
 
 if(WIN32)
   string(REPLACE "\\" "/" TENSORRT_ROOT "${TENSORRT_ROOT}")
-  set(TR_INFER_LIB nvinfer.lib)
-  set(TR_INFER_RT nvinfer.dll)
-  set(TR_INFER_PLUGIN_RT nvinfer_plugin.dll)
+  string(REGEX MATCH "([0-9]+)\\.([0-9]+)\\.([0-9]+)\\.([0-9]+)"
+               TENSORRT_VERSION ${TENSORRT_ROOT})
+  if(NOT TENSORRT_VERSION)
+    set(MAJOR_VERSION "unknown")
+  else()
+    message(STATUS "TensorRT Full Version: ${TENSORRT_VERSION}")
+    string(REGEX MATCH "^[0-9]+" MAJOR_VERSION ${TENSORRT_VERSION})
+  endif()
+  if(MAJOR_VERSION STREQUAL "10")
+    message(STATUS "TensorRT version is 10, applying specific settings.")
+    set(TR_INFER_LIB nvinfer_10.lib)
+    set(TR_INFER_RT nvinfer_10.dll)
+    set(TR_INFER_PLUGIN_RT nvinfer_plugin_10.dll)
+  else()
+    message(STATUS "TensorRT version is not 10, using default settings.")
+    set(TR_INFER_LIB nvinfer.lib)
+    set(TR_INFER_RT nvinfer.dll)
+    set(TR_INFER_PLUGIN_RT nvinfer_plugin.dll)
+  endif()
 else()
   set(TENSORRT_ROOT
       "/usr"
diff --git a/python/setup.py.in b/python/setup.py.in
@@ -656,6 +656,15 @@ def get_paddle_extra_install_requirements():
                     "nvidia-cusolver-cu12==11.7.1.2 | "
                     "nvidia-cusparse-cu12==12.5.4.2 "
                 ),
+                "12.8": (
+                    "nvidia-cuda-runtime-cu12==12.8.57 | "
+                    "nvidia-cudnn-cu12==9.7.1.26 | "
+                    "nvidia-cublas-cu12==12.8.3.14 | "
+                    "nvidia-cufft-cu12==11.3.3.41 | "
+                    "nvidia-curand-cu12==10.3.9.55 | "
+                    "nvidia-cusolver-cu12==11.7.2.55 | "
+                    "nvidia-cusparse-cu12==12.5.7.53 "
+                ),
             }
 
         try:
diff --git a/setup.py b/setup.py
@@ -1175,6 +1175,15 @@ def get_paddle_extra_install_requirements():
                     "nvidia-cusolver-cu12==11.7.1.2 | "
                     "nvidia-cusparse-cu12==12.5.4.2 "
                 ),
+                "12.8": (
+                    "nvidia-cuda-runtime-cu12==12.8.57 | "
+                    "nvidia-cudnn-cu12==9.7.1.26 | "
+                    "nvidia-cublas-cu12==12.8.3.14 | "
+                    "nvidia-cufft-cu12==11.3.3.41 | "
+                    "nvidia-curand-cu12==10.3.9.55 | "
+                    "nvidia-cusolver-cu12==11.7.2.55 | "
+                    "nvidia-cusparse-cu12==12.5.7.53 "
+                ),
             }
         try:
             output = subprocess.check_output(['nvcc', '--version']).decode(
