From 62b6a9399de913cd83a45bb52b6bb3444e59a23b Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Sun, 9 Oct 2022 19:02:28 -0700 Subject: Added CUDA 11.8 install and deployment. --- cuda_install.sh | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'cuda_install.sh') diff --git a/cuda_install.sh b/cuda_install.sh index 4508f4a..b6c553b 100644 --- a/cuda_install.sh +++ b/cuda_install.sh @@ -10,6 +10,7 @@ URL114=https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installer URL115=https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux.run URL116=https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run URL117=https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run +URL118=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run CUDA_VERSION=$1 @@ -52,6 +53,9 @@ if [[ -n "$CUDA_VERSION" ]]; then elif [[ "$CUDA_VERSION" -eq "117" ]]; then URL=$URL117 FOLDER=cuda-11.7 + elif [[ "$CUDA_VERSION" -eq "118" ]]; then + URL=$URL118 + FOLDER=cuda-11.8 else echo "argument error: No cuda version passed as input. Choose among: {111, 115}" fi -- cgit v1.2.3