@@ -44,8 +44,8 @@ function install_118 {
44
44
rm -rf tmp_cudnn
45
45
46
46
# 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
49
49
cd nccl && make -j src.build
50
50
cp -a build/include/* /usr/local/cuda/include/
51
51
cp -a build/lib/* /usr/local/cuda/lib64/
@@ -58,7 +58,7 @@ function install_118 {
58
58
}
59
59
60
60
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"
62
62
rm -rf /usr/local/cuda-12.1 /usr/local/cuda
63
63
# install CUDA 12.1.0 in the same container
64
64
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 {
77
77
rm -rf tmp_cudnn
78
78
79
79
# 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
82
82
cd nccl && make -j src.build
83
83
cp -a build/include/* /usr/local/cuda/include/
84
84
cp -a build/lib/* /usr/local/cuda/lib64/
@@ -91,7 +91,7 @@ function install_121 {
91
91
}
92
92
93
93
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"
95
95
rm -rf /usr/local/cuda-12.4 /usr/local/cuda
96
96
# install CUDA 12.4.0 in the same container
97
97
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 {
110
110
rm -rf tmp_cudnn
111
111
112
112
# 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
115
115
cd nccl && make -j src.build
116
116
cp -a build/include/* /usr/local/cuda/include/
117
117
cp -a build/lib/* /usr/local/cuda/lib64/
0 commit comments