22
33set -ex
44
5+ CUDNN_VERSION=9.1.0.70
6+
57function install_cusparselt_040 {
68 # cuSparseLt license: https://docs.nvidia.com/cuda/cusparselt/license.html
79 mkdir tmp_cusparselt && pushd tmp_cusparselt
@@ -25,7 +27,7 @@ function install_cusparselt_052 {
2527}
2628
2729function install_118 {
28- echo " Installing CUDA 11.8 and cuDNN 8.7 and NCCL 2.15 and cuSparseLt-0.4.0"
30+ echo " Installing CUDA 11.8 and cuDNN ${CUDNN_VERSION} and NCCL 2.15 and cuSparseLt-0.4.0"
2931 rm -rf /usr/local/cuda-11.8 /usr/local/cuda
3032 # install CUDA 11.8.0 in the same container
3133 wget -q https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
@@ -36,10 +38,10 @@ function install_118 {
3638
3739 # cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
3840 mkdir tmp_cudnn && cd tmp_cudnn
39- wget -q https://developer.download.nvidia.com/compute/redist/cudnn/v8.7.0/local_installers/11.8/ cudnn-linux-x86_64-8.7.0.84_cuda11 -archive.tar.xz -O cudnn-linux-x86_64-8.7.0.84_cuda11 -archive.tar.xz
40- tar xf cudnn-linux-x86_64-8.7.0.84_cuda11 -archive.tar.xz
41- cp -a cudnn-linux-x86_64-8.7.0.84_cuda11 -archive/include/* /usr/local/cuda/include/
42- cp -a cudnn-linux-x86_64-8.7.0.84_cuda11 -archive/lib/* /usr/local/cuda/lib64/
41+ wget -q https://developer.download.nvidia.com/compute/cudnn/ redist/cudnn/linux-x86_64/ cudnn-linux-x86_64-${CUDNN_VERSION} _cuda11 -archive.tar.xz -O cudnn-linux-x86_64-${CUDNN_VERSION} _cuda11 -archive.tar.xz
42+ tar xf cudnn-linux-x86_64-${CUDNN_VERSION} _cuda11 -archive.tar.xz
43+ cp -a cudnn-linux-x86_64-${CUDNN_VERSION} _cuda11 -archive/include/* /usr/local/cuda/include/
44+ cp -a cudnn-linux-x86_64-${CUDNN_VERSION} _cuda11 -archive/lib/* /usr/local/cuda/lib64/
4345 cd ..
4446 rm -rf tmp_cudnn
4547
@@ -58,7 +60,7 @@ function install_118 {
5860}
5961
6062function install_121 {
61- echo " Installing CUDA 12.1 and cuDNN 8.9 and NCCL 2.20.5 and cuSparseLt-0.5.2"
63+ echo " Installing CUDA 12.1 and cuDNN ${CUDNN_VERSION} and NCCL 2.20.5 and cuSparseLt-0.5.2"
6264 rm -rf /usr/local/cuda-12.1 /usr/local/cuda
6365 # install CUDA 12.1.0 in the same container
6466 wget -q https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
@@ -69,10 +71,10 @@ function install_121 {
6971
7072 # cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
7173 mkdir tmp_cudnn && cd tmp_cudnn
72- wget -q https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.2.26_cuda12 -archive.tar.xz -O cudnn-linux-x86_64-8.9.2.26_cuda12 -archive.tar.xz
73- tar xf cudnn-linux-x86_64-8.9.2.26_cuda12 -archive.tar.xz
74- cp -a cudnn-linux-x86_64-8.9.2.26_cuda12 -archive/include/* /usr/local/cuda/include/
75- cp -a cudnn-linux-x86_64-8.9.2.26_cuda12 -archive/lib/* /usr/local/cuda/lib64/
74+ wget -q https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive.tar.xz -O cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive.tar.xz
75+ tar xf cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive.tar.xz
76+ cp -a cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive/include/* /usr/local/cuda/include/
77+ cp -a cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive/lib/* /usr/local/cuda/lib64/
7678 cd ..
7779 rm -rf tmp_cudnn
7880
@@ -91,7 +93,7 @@ function install_121 {
9193}
9294
9395function install_124 {
94- echo " Installing CUDA 12.4 and cuDNN 8.9 and NCCL 2.20.5 and cuSparseLt-0.5.2"
96+ echo " Installing CUDA 12.4 and cuDNN ${CUDNN_VERSION} and NCCL 2.20.5 and cuSparseLt-0.5.2"
9597 rm -rf /usr/local/cuda-12.4 /usr/local/cuda
9698 # install CUDA 12.4.0 in the same container
9799 wget -q https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
@@ -102,10 +104,10 @@ function install_124 {
102104
103105 # cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
104106 mkdir tmp_cudnn && cd tmp_cudnn
105- wget -q https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.2.26_cuda12 -archive.tar.xz -O cudnn-linux-x86_64-8.9.2.26_cuda12 -archive.tar.xz
106- tar xf cudnn-linux-x86_64-8.9.2.26_cuda12 -archive.tar.xz
107- cp -a cudnn-linux-x86_64-8.9.2.26_cuda12 -archive/include/* /usr/local/cuda/include/
108- cp -a cudnn-linux-x86_64-8.9.2.26_cuda12 -archive/lib/* /usr/local/cuda/lib64/
107+ wget -q https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive.tar.xz -O cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive.tar.xz
108+ tar xf cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive.tar.xz
109+ cp -a cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive/include/* /usr/local/cuda/include/
110+ cp -a cudnn-linux-x86_64-${CUDNN_VERSION} _cuda12 -archive/lib/* /usr/local/cuda/lib64/
109111 cd ..
110112 rm -rf tmp_cudnn
111113
0 commit comments