Skip to content

Commit 9b0d3f9

Browse files
committed
Update nccl build version
1 parent 48c6f79 commit 9b0d3f9

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

common/install_cuda.sh

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ function install_118 {
4444
rm -rf tmp_cudnn
4545

4646
# NCCL license: https://docs.nvidia.com/deeplearning/nccl/#licenses
47-
# Follow build: https://github.com/NVIDIA/nccl/tree/v2.19.3-1?tab=readme-ov-file#build
48-
git clone -b v2.19.3-1 --depth 1 https://github.com/NVIDIA/nccl.git
47+
# Follow build: https://github.com/NVIDIA/nccl/tree/master?tab=readme-ov-file#build
48+
git clone -b v2.20.5-1 --depth 1 https://github.com/NVIDIA/nccl.git
4949
cd nccl && make -j src.build
5050
cp -a build/include/* /usr/local/cuda/include/
5151
cp -a build/lib/* /usr/local/cuda/lib64/
@@ -58,7 +58,7 @@ function install_118 {
5858
}
5959

6060
function install_121 {
61-
echo "Installing CUDA 12.1 and cuDNN 8.9 and NCCL 2.19.3 and cuSparseLt-0.5.2"
61+
echo "Installing CUDA 12.1 and cuDNN 8.9 and NCCL 2.20.5 and cuSparseLt-0.5.2"
6262
rm -rf /usr/local/cuda-12.1 /usr/local/cuda
6363
# install CUDA 12.1.0 in the same container
6464
wget -q https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
@@ -77,8 +77,8 @@ function install_121 {
7777
rm -rf tmp_cudnn
7878

7979
# NCCL license: https://docs.nvidia.com/deeplearning/nccl/#licenses
80-
# Follow build: https://github.com/NVIDIA/nccl/tree/v2.19.3-1?tab=readme-ov-file#build
81-
git clone -b v2.19.3-1 --depth 1 https://github.com/NVIDIA/nccl.git
80+
# Follow build: https://github.com/NVIDIA/nccl/tree/master?tab=readme-ov-file#build
81+
git clone -b v2.20.5-1 --depth 1 https://github.com/NVIDIA/nccl.git
8282
cd nccl && make -j src.build
8383
cp -a build/include/* /usr/local/cuda/include/
8484
cp -a build/lib/* /usr/local/cuda/lib64/
@@ -91,7 +91,7 @@ function install_121 {
9191
}
9292

9393
function install_124 {
94-
echo "Installing CUDA 12.4 and cuDNN 8.9 and NCCL 2.19.3 and cuSparseLt-0.5.2"
94+
echo "Installing CUDA 12.4 and cuDNN 8.9 and NCCL 2.20.5 and cuSparseLt-0.5.2"
9595
rm -rf /usr/local/cuda-12.4 /usr/local/cuda
9696
# install CUDA 12.4.0 in the same container
9797
wget -q https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
@@ -110,8 +110,8 @@ function install_124 {
110110
rm -rf tmp_cudnn
111111

112112
# NCCL license: https://docs.nvidia.com/deeplearning/nccl/#licenses
113-
# Follow build: https://github.com/NVIDIA/nccl/tree/v2.19.3-1?tab=readme-ov-file#build
114-
git clone -b v2.19.3-1 --depth 1 https://github.com/NVIDIA/nccl.git
113+
# Follow build: https://github.com/NVIDIA/nccl/tree/master?tab=readme-ov-file#build
114+
git clone -b v2.20.5-1 --depth 1 https://github.com/NVIDIA/nccl.git
115115
cd nccl && make -j src.build
116116
cp -a build/include/* /usr/local/cuda/include/
117117
cp -a build/lib/* /usr/local/cuda/lib64/

0 commit comments

Comments
 (0)