diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-25 14:02:14 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-25 14:02:14 -0700 |
| commit | 8b1fd32e3e4f5073fd055cb5f9261ec585f8cc2c (patch) | |
| tree | 76044424d73d02e1026c996b22b9da5061188387 /cuda_install_111.sh | |
| parent | 7d2ecd30c044840ba5f161ec73e5eaf30ac8131d (diff) | |
Fixed makefile; fixed Ampere igemmlt_8 bug.
Diffstat (limited to 'cuda_install_111.sh')
| -rw-r--r-- | cuda_install_111.sh | 38 |
1 files changed, 38 insertions, 0 deletions
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 + + + |