File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -37,10 +37,10 @@ function install_121 {
37
37
echo " Installing CUDA 12.1 and cuDNN 8.9 and NCCL 2.18.1"
38
38
rm -rf /usr/local/cuda-12.1 /usr/local/cuda
39
39
# install CUDA 12.1.0 in the same container
40
- wget -q https://developer.download.nvidia.com/compute/cuda/12.1.0 /local_installers/cuda_12.1.0_530 .30.02_linux.run
41
- chmod +x cuda_12.1.0_530 .30.02_linux.run
42
- ./cuda_12.1.0_530 .30.02_linux.run --toolkit --silent
43
- rm -f cuda_12.1.0_530 .30.02_linux.run
40
+ wget -q https://developer.download.nvidia.com/compute/cuda/12.1.1 /local_installers/cuda_12.1.1_530 .30.02_linux.run
41
+ chmod +x cuda_12.1.1_530 .30.02_linux.run
42
+ ./cuda_12.1.1_530 .30.02_linux.run --toolkit --silent
43
+ rm -f cuda_12.1.1_530 .30.02_linux.run
44
44
rm -f /usr/local/cuda && ln -s /usr/local/cuda-12.1 /usr/local/cuda
45
45
46
46
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
You can’t perform that action at this time.
0 commit comments