Skip to content

GitLab

"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "ebe3f023c4004cef103ee0a3bfaf0898cccd4766"
Unverified Commit e048dc92 authored by theamdara's avatar theamdara Committed by GitHub
Browse files

Update cuda_install.sh 

Add option to download and install Cuda 12.3
parent 726f1470
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
cuda_install.sh
View file @ e048dc92
......@@ -14,6 +14,7 @@ URL118=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installer
URL120=https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run
URL121=https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
URL122=https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run
URL123=https://developer.download.nvidia.com/compute/cuda/12.3.1/local_installers/cuda_12.3.1_545.23.08_linux.run
CUDA_VERSION=$1
......@@ -69,6 +70,9 @@ if [[ -n "$CUDA_VERSION" ]]; then
elif [[ "$CUDA_VERSION" -eq "122" ]]; then
URL=$URL122
FOLDER=cuda-12.2
elif [[ "$CUDA_VERSION" -eq "123" ]]; then
URL=$URL123
FOLDER=cuda-12.3
else
echo "argument error: No cuda version passed as input. Choose among versions 92 to 121"
fi
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment