diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-10-09 19:02:28 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-10-09 19:02:28 -0700 |
| commit | 62b6a9399de913cd83a45bb52b6bb3444e59a23b (patch) | |
| tree | 5520e16d20408eddcceee027335d9b733dbedffb /cuda_install.sh | |
| parent | ed2e3b9db4039272af9e3ef13f73f92f811c8a81 (diff) | |
Added CUDA 11.8 install and deployment.
Diffstat (limited to 'cuda_install.sh')
| -rw-r--r-- | cuda_install.sh | 4 |
1 files changed, 4 insertions, 0 deletions
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 |