Skip to content

GitLab

"...python/git@developer.sourcefind.cn:change/sglang.git" did not exist on "da3890e82a97e9b1f970133c0fef6341a8bbf791"
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
Show 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 ...@@ -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 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 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 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 CUDA_VERSION=$1
...@@ -69,6 +70,9 @@ if [[ -n "$CUDA_VERSION" ]]; then ...@@ -69,6 +70,9 @@ if [[ -n "$CUDA_VERSION" ]]; then
elif [[ "$CUDA_VERSION" -eq "122" ]]; then elif [[ "$CUDA_VERSION" -eq "122" ]]; then
URL=$URL122 URL=$URL122
FOLDER=cuda-12.2 FOLDER=cuda-12.2
elif [[ "$CUDA_VERSION" -eq "123" ]]; then
URL=$URL123
FOLDER=cuda-12.3
else else
echo "argument error: No cuda version passed as input. Choose among versions 92 to 121" echo "argument error: No cuda version passed as input. Choose among versions 92 to 121"
fi 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