Skip to content

Commit 941be28

Browse files
authored
update to CUDA 12.1U1 (#1476)
Should fix pytorch/pytorch#94772 in wheel builds
1 parent 912957a commit 941be28

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

common/install_cuda.sh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,10 @@ function install_121 {
3737
echo "Installing CUDA 12.1 and cuDNN 8.9 and NCCL 2.18.1"
3838
rm -rf /usr/local/cuda-12.1 /usr/local/cuda
3939
# 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
4444
rm -f /usr/local/cuda && ln -s /usr/local/cuda-12.1 /usr/local/cuda
4545

4646
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement

0 commit comments

Comments
 (0)