diff options
| -rw-r--r-- | bitsandbytes/cuda_setup/main.py | 8 | ||||
| -rw-r--r-- | cuda_install.sh | 2 | ||||
| -rw-r--r-- | deploy_from_slurm.sh | 2 | ||||
| -rw-r--r-- | setup.py | 2 |
4 files changed, 11 insertions, 3 deletions
diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py index f1c845c..1f2ceb4 100644 --- a/bitsandbytes/cuda_setup/main.py +++ b/bitsandbytes/cuda_setup/main.py @@ -45,6 +45,9 @@ def get_cuda_version(cuda, cudart_path): major = version//1000 minor = (version-(major*1000))//10 + if major < 11: + print('CUDA SETUP: CUDA version lower than 11 are currenlty not supported!') + return f'{major}{minor}' @@ -110,6 +113,10 @@ def get_compute_capability(cuda): def evaluate_cuda_setup(): + print('') + print('='*35 + 'BUG REPORT' + '='*35) + print('Welcome to bitsandbytes. For bug reports, please use this form: https://docs.google.com/forms/d/e/1FAIpQLScPB8emS3Thkp66nvqwmjTEgxp8Y9ufuWTzFyr9kJ5AoI47dQ/viewform?usp=sf_link') + print('='*80) binary_name = "libbitsandbytes_cpu.so" cudart_path = determine_cuda_runtime_lib_path() if cudart_path is None: @@ -121,6 +128,7 @@ def evaluate_cuda_setup(): print(f"CUDA SETUP: CUDA path found: {cudart_path}") cuda = get_cuda_lib_handle() cc = get_compute_capability(cuda) + print(f"CUDA SETUP: Highest compute capability among GPUs detected: {cc}") cuda_version_string = get_cuda_version(cuda, cudart_path) diff --git a/cuda_install.sh b/cuda_install.sh index 856cbe5..4508f4a 100644 --- a/cuda_install.sh +++ b/cuda_install.sh @@ -65,7 +65,7 @@ if [[ -n "$CUDA_VERSION" ]]; then echo $URL echo $FILE wget $URL - bash $FILE --no-drm --no-man-page --override --installpath=~/local --librarypath=$BASE_PATH/lib --toolkitpath=$BASE_PATH/$FOLDER/ --toolkit --silent + 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 diff --git a/deploy_from_slurm.sh b/deploy_from_slurm.sh index c6ee84d..bc53869 100644 --- a/deploy_from_slurm.sh +++ b/deploy_from_slurm.sh @@ -202,4 +202,4 @@ if [ ! -f "./bitsandbytes/libbitsandbytes_cuda117_nocublaslt.so" ]; then fi python -m build -python -m twine upload dist/* --verbose --repository testpypi +python -m twine upload dist/* --verbose @@ -18,7 +18,7 @@ def read(fname): setup( name=f"bitsandbytes", - version=f"0.31.7", + version=f"0.31.8", author="Tim Dettmers", author_email="dettmers@cs.washington.edu", description="8-bit optimizers and matrix multiplication routines.", |