diff options
-rw-r--r-- | cuda_install.sh | 4 | ||||
-rw-r--r-- | deploy.sh (renamed from deploy_from_slurm.sh) | 20 |
2 files changed, 24 insertions, 0 deletions
diff --git a/cuda_install.sh b/cuda_install.sh index 4508f4a..b6c553b 100644 --- a/cuda_install.sh +++ b/cuda_install.sh @@ -10,6 +10,7 @@ URL114=https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installer 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 +URL118=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run CUDA_VERSION=$1 @@ -52,6 +53,9 @@ if [[ -n "$CUDA_VERSION" ]]; then elif [[ "$CUDA_VERSION" -eq "117" ]]; then URL=$URL117 FOLDER=cuda-11.7 + elif [[ "$CUDA_VERSION" -eq "118" ]]; then + URL=$URL118 + FOLDER=cuda-11.8 else echo "argument error: No cuda version passed as input. Choose among: {111, 115}" fi diff --git a/deploy_from_slurm.sh b/deploy.sh index bc53869..7d4f1a0 100644 --- a/deploy_from_slurm.sh +++ b/deploy.sh @@ -108,6 +108,16 @@ if [ ! -f "./bitsandbytes/libbitsandbytes_cuda117.so" ]; then exit 64 fi +make clean +export CUDA_HOME=$BASE_PATH/cuda-11.8 +make cuda11x CUDA_VERSION=118 + +if [ ! -f "./bitsandbytes/libbitsandbytes_cuda118.so" ]; then + # Control will enter here if $DIRECTORY doesn't exist. + echo "Compilation unsuccessul!" 1>&2 + exit 64 +fi + make clean export CUDA_HOME=$BASE_PATH/cuda-10.2 @@ -201,5 +211,15 @@ if [ ! -f "./bitsandbytes/libbitsandbytes_cuda117_nocublaslt.so" ]; then exit 64 fi +make clean +export CUDA_HOME=$BASE_PATH/cuda-11.8 +make cuda11x_nomatmul CUDA_VERSION=118 + +if [ ! -f "./bitsandbytes/libbitsandbytes_cuda118_nocublaslt.so" ]; then + # Control will enter here if $DIRECTORY doesn't exist. + echo "Compilation unsuccessul!" 1>&2 + exit 64 +fi + python -m build python -m twine upload dist/* --verbose |