@@ -93,36 +93,36 @@ function install_121 {
9393}
9494
9595function install_124 {
96- echo " Installing CUDA 12.4 and cuDNN ${CUDNN_VERSION} and NCCL 2.20.5 and cuSparseLt-0.5.2"
97- rm -rf /usr/local/cuda-12.4 /usr/local/cuda
98- # install CUDA 12.4.0 in the same container
99- wget -q https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
100- chmod +x cuda_12.4.0_550.54.14_linux.run
101- ./cuda_12.4.0_550.54.14_linux.run --toolkit --silent
102- rm -f cuda_12.4.0_550.54.14_linux.run
103- rm -f /usr/local/cuda && ln -s /usr/local/cuda-12.4 /usr/local/cuda
104-
105- # cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
106- mkdir tmp_cudnn && cd tmp_cudnn
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/
111- cd ..
112- rm -rf tmp_cudnn
113-
114- # NCCL license: https://docs.nvidia.com/deeplearning/nccl/#licenses
115- # Follow build: https://github.com/NVIDIA/nccl/tree/master?tab=readme-ov-file#build
116- git clone -b v2.20.5-1 --depth 1 https://github.com/NVIDIA/nccl.git
117- cd nccl && make -j src.build
118- cp -a build/include/* /usr/local/cuda/include/
119- cp -a build/lib/* /usr/local/cuda/lib64/
96+ echo " Installing CUDA 12.4 and cuDNN ${CUDNN_VERSION} and NCCL 2.20.5 and cuSparseLt-0.5.2"
97+ rm -rf /usr/local/cuda-12.4 /usr/local/cuda
98+ # install CUDA 12.4.0 in the same container
99+ wget -q https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
100+ chmod +x cuda_12.4.0_550.54.14_linux.run
101+ ./cuda_12.4.0_550.54.14_linux.run --toolkit --silent
102+ rm -f cuda_12.4.0_550.54.14_linux.run
103+ rm -f /usr/local/cuda && ln -s /usr/local/cuda-12.4 /usr/local/cuda
104+
105+ # cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
106+ mkdir tmp_cudnn && cd tmp_cudnn
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/
120111 cd ..
121- rm -rf nccl
112+ rm -rf tmp_cudnn
122113
123- install_cusparselt_052
114+ # NCCL license: https://docs.nvidia.com/deeplearning/nccl/#licenses
115+ # Follow build: https://github.com/NVIDIA/nccl/tree/master?tab=readme-ov-file#build
116+ git clone -b v2.20.5-1 --depth 1 https://github.com/NVIDIA/nccl.git
117+ cd nccl && make -j src.build
118+ cp -a build/include/* /usr/local/cuda/include/
119+ cp -a build/lib/* /usr/local/cuda/lib64/
120+ cd ..
121+ rm -rf nccl
124122
125- ldconfig
123+ install_cusparselt_052
124+
125+ ldconfig
126126}
127127
128128function prune_118 {
0 commit comments