From 8b1fd32e3e4f5073fd055cb5f9261ec585f8cc2c Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Mon, 25 Jul 2022 14:02:14 -0700 Subject: Fixed makefile; fixed Ampere igemmlt_8 bug. --- cuda_install_111.sh | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 cuda_install_111.sh (limited to 'cuda_install_111.sh') diff --git a/cuda_install_111.sh b/cuda_install_111.sh new file mode 100644 index 0000000..476ab59 --- /dev/null +++ b/cuda_install_111.sh @@ -0,0 +1,38 @@ +FILE115=:cuda_11.5.1_495.29.05_linux.run +FILE111=:cuda_11.1.1_455.32.00_linux.run +URL115=:https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux.run +URL111=:https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run + + +CUDA_VERSION=$1 + +if [[ -n "$CUDA_VERSION" ]]; then + if [[ "$CUDA_VERSION" -eq "111" ]]; then + FILE=cuda_11.1.1_455.32.00_linux.run + URL=https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run + FOLDER=cuda-11.1 + elif [[ "$CUDA_VERSION" -eq "115" ]]; then + FILE=cuda_11.5.1_495.29.05_linux.run + URL=https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux.run + FOLDER=cuda-11.5 + 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 + +if [[ -n "$CUDA_VERSION" ]]; then + echo $URL + echo $FILE + wget $URL + bash $FILE --no-drm --no-man-page --override --installpath=~/local --librarypath=~/local/lib --toolkitpath=~/local/$FOLDER/ --toolkit --silent + echo "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/local/$FOLDER/lib64/" >> ~/.bashrc + echo "export PATH=$PATH:~/local/$FOLDER/bin/" >> ~/.bashrc + source ~/.bashrc +else + echo "" +fi + + + -- cgit v1.2.3