[Setup] Update nvidia-cudnn-cu12 version from 9.0.0.312 (#69349)

HydrogenSulfate · web-flow · commit 2ef18ba97e57 · 2024-11-14T16:18:35.000+08:00
* update nvidia-cudnn-cu12 version from 9.0.0.312
9.0.0.312 to 9.1.1.17

* update more files
diff --git a/python/setup.py.in b/python/setup.py.in
@@ -569,7 +569,7 @@ def get_paddle_extra_install_requirements():
             "V12": (
                 "nvidia-cuda-runtime-cu12==12.3.101; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                 "nvidia-cuda-cupti-cu12==12.3.101; platform_system == 'Linux' and platform_machine == 'x86_64' | "
-                "nvidia-cudnn-cu12==9.0.0.312; platform_system == 'Linux' and platform_machine == 'x86_64' | "
+                "nvidia-cudnn-cu12==9.1.1.17; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                 "nvidia-cublas-cu12==12.3.4.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                 "nvidia-cufft-cu12==11.2.1.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                 "nvidia-curand-cu12==10.3.5.147; platform_system == 'Linux' and platform_machine == 'x86_64' | "
@@ -593,7 +593,7 @@ def get_paddle_extra_install_requirements():
                 ),
                 "V12": (
                     "nvidia-cuda-runtime-cu12==12.3.101 | "
-                    "nvidia-cudnn-cu12==9.0.0.312 | "
+                    "nvidia-cudnn-cu12==9.1.1.17 | "
                     "nvidia-cublas-cu12==12.3.4.1 | "
                     "nvidia-cufft-cu12==11.2.1.3 | "
                     "nvidia-curand-cu12==10.3.5.147 | "
diff --git a/python/setup_cinn.py.in b/python/setup_cinn.py.in
@@ -208,7 +208,7 @@ if platform.system() == 'Linux' and platform.machine() == 'x86_64':
             "V12": (
                 "nvidia-cuda-runtime-cu12==12.3.101; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                 "nvidia-cuda-cupti-cu12==12.3.101; platform_system == 'Linux' and platform_machine == 'x86_64' | "
-                "nvidia-cudnn-cu12==9.0.0.312; platform_system == 'Linux' and platform_machine == 'x86_64' | "
+                "nvidia-cudnn-cu12==9.1.1.17; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                 "nvidia-cublas-cu12==12.3.4.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                 "nvidia-cufft-cu12==11.2.1.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                 "nvidia-curand-cu12==10.3.5.147; platform_system == 'Linux' and platform_machine == 'x86_64' | "
diff --git a/setup.py b/setup.py
@@ -1088,7 +1088,7 @@ def get_paddle_extra_install_requirements():
                 "V12": (
                     "nvidia-cuda-runtime-cu12==12.3.101; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                     "nvidia-cuda-cupti-cu12==12.3.101; platform_system == 'Linux' and platform_machine == 'x86_64' | "
-                    "nvidia-cudnn-cu12==9.0.0.312; platform_system == 'Linux' and platform_machine == 'x86_64' | "
+                    "nvidia-cudnn-cu12==9.1.1.17; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                     "nvidia-cublas-cu12==12.3.4.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                     "nvidia-cufft-cu12==11.2.1.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
                     "nvidia-curand-cu12==10.3.5.147; platform_system == 'Linux' and platform_machine == 'x86_64' | "
@@ -1112,7 +1112,7 @@ def get_paddle_extra_install_requirements():
                 ),
                 "V12": (
                     "nvidia-cuda-runtime-cu12==12.3.101 | "
-                    "nvidia-cudnn-cu12==9.0.0.312 | "
+                    "nvidia-cudnn-cu12==9.1.1.17 | "
                     "nvidia-cublas-cu12==12.3.4.1 | "
                     "nvidia-cufft-cu12==11.2.1.3 | "
                     "nvidia-curand-cu12==10.3.5.147 | "
diff --git a/tools/dockerfile/build_scripts/install_cudnn.sh b/tools/dockerfile/build_scripts/install_cudnn.sh
@@ -87,15 +87,15 @@ elif [[ "$1" == "cudnn896" ]]; then
   rm -f cudnn-linux-x86_64-8.9.6.50_cuda12-archive.tar.xz && \
   rm -rf cudnn-linux-x86_64-8.9.6.50_cuda12-archive
 elif [[ "$1" == "cudnn900" ]]; then
-  wget -q https://paddle-ci.gz.bcebos.com/cudnn/cudnn-linux-x86_64-9.0.0.312_cuda12-archive.tar.xz --no-check-certificate
-  tar xJvf cudnn-linux-x86_64-9.0.0.312_cuda12-archive.tar.xz && \
-  cd cudnn-linux-x86_64-9.0.0.312_cuda12-archive && \
+  wget -q https://paddle-ci.gz.bcebos.com/cudnn/cudnn-linux-x86_64-9.1.1.17_cuda12-archive.tar.xz --no-check-certificate
+  tar xJvf cudnn-linux-x86_64-9.1.1.17_cuda12-archive.tar.xz && \
+  cd cudnn-linux-x86_64-9.1.1.17_cuda12-archive && \
   cp -r include /usr && \
   mkdir -p /usr/lib/x86_64-linux-gnu && \
   cp -r lib/libcudnn* /usr/lib/x86_64-linux-gnu && \
   cp -r lib /usr && cd ../ && \
-  rm -f cudnn-linux-x86_64-9.0.0.312_cuda12-archive.tar.xz && \
-  rm -rf cudnn-linux-x86_64-9.0.0.312_cuda12-archive
+  rm -f cudnn-linux-x86_64-9.1.1.17_cuda12-archive.tar.xz && \
+  rm -rf cudnn-linux-x86_64-9.1.1.17_cuda12-archive
 elif [[ "$1" == "cudnn911" ]]; then
   wget -q https://paddle-ci.gz.bcebos.com/cudnn/cudnn-linux-x86_64-9.1.1.17_cuda12-archive.tar.xz --no-check-certificate
   tar xJvf cudnn-linux-x86_64-9.1.1.17_cuda12-archive.tar.xz && \
