Skip to content

Commit 7108860

Browse files
committed
Remove 2 spaces for install_cu124
1 parent 7e970d7 commit 7108860

File tree

1 file changed

+27
-27
lines changed

1 file changed

+27
-27
lines changed

common/install_cuda.sh

+27-27
Original file line numberDiff line numberDiff line change
@@ -93,36 +93,36 @@ function install_121 {
9393
}
9494

9595
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/
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

128128
function prune_118 {

0 commit comments

Comments
 (0)