diff options
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 |