forked from mrq/bitsandbytes-rocm
78 lines
2.9 KiB
Bash
78 lines
2.9 KiB
Bash
URL92=https://developer.nvidia.com/compute/cuda/9.2/Prod2/local_installers/cuda_9.2.148_396.37_linux
|
|
URL100=https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux
|
|
URL101=https://developer.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.105_418.39_linux.run
|
|
URL102=https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
|
|
URL110=https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
|
|
URL111=https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
|
|
URL112=https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux.run
|
|
URL113=https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
|
|
URL114=https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run
|
|
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
|
|
|
|
|
|
CUDA_VERSION=$1
|
|
BASE_PATH=$2
|
|
|
|
if [[ -n "$CUDA_VERSION" ]]; then
|
|
if [[ "$CUDA_VERSION" -eq "92" ]]; then
|
|
URL=$URL92
|
|
FOLDER=cuda-9.2
|
|
elif [[ "$CUDA_VERSION" -eq "100" ]]; then
|
|
URL=$URL100
|
|
FOLDER=cuda-10.0
|
|
elif [[ "$CUDA_VERSION" -eq "101" ]]; then
|
|
URL=$URL101
|
|
FOLDER=cuda-10.1
|
|
elif [[ "$CUDA_VERSION" -eq "102" ]]; then
|
|
URL=$URL102
|
|
FOLDER=cuda-10.2
|
|
elif [[ "$CUDA_VERSION" -eq "110" ]]; then
|
|
URL=$URL110
|
|
FOLDER=cuda-11.0
|
|
elif [[ "$CUDA_VERSION" -eq "111" ]]; then
|
|
URL=$URL111
|
|
FOLDER=cuda-11.1
|
|
elif [[ "$CUDA_VERSION" -eq "112" ]]; then
|
|
URL=$URL112
|
|
FOLDER=cuda-11.2
|
|
elif [[ "$CUDA_VERSION" -eq "113" ]]; then
|
|
URL=$URL113
|
|
FOLDER=cuda-11.3
|
|
elif [[ "$CUDA_VERSION" -eq "114" ]]; then
|
|
URL=$URL114
|
|
FOLDER=cuda-11.4
|
|
elif [[ "$CUDA_VERSION" -eq "115" ]]; then
|
|
URL=$URL115
|
|
FOLDER=cuda-11.5
|
|
elif [[ "$CUDA_VERSION" -eq "116" ]]; then
|
|
URL=$URL116
|
|
FOLDER=cuda-11.6
|
|
elif [[ "$CUDA_VERSION" -eq "117" ]]; then
|
|
URL=$URL117
|
|
FOLDER=cuda-11.7
|
|
else
|
|
echo "argument error: No cuda version passed as input. Choose among: {111, 115}"
|
|
fi
|
|
else
|
|
echo "argument error: No cuda version passed as input. Choose among: {111, 115}"
|
|
fi
|
|
|
|
FILE=$(basename $URL)
|
|
|
|
if [[ -n "$CUDA_VERSION" ]]; then
|
|
echo $URL
|
|
echo $FILE
|
|
wget $URL
|
|
bash $FILE --no-drm --no-man-page --override --toolkitpath=$BASE_PATH/$FOLDER/ --toolkit --silent
|
|
echo "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$BASE_PATH/$FOLDER/lib64/" >> ~/.bashrc
|
|
echo "export PATH=$PATH:$BASE_PATH/$FOLDER/bin/" >> ~/.bashrc
|
|
source ~/.bashrc
|
|
else
|
|
echo ""
|
|
fi
|
|
|
|
|
|
|