@@ -93,36 +93,36 @@ function install_121 {
93
93
}
94
94
95
95
function 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/
120
111
cd ..
121
- rm -rf nccl
112
+ rm -rf tmp_cudnn
122
113
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
124
122
125
- ldconfig
123
+ install_cusparselt_052
124
+
125
+ ldconfig
126
126
}
127
127
128
128
function prune_118 {
0 commit comments