mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues1 Pull Requests Packages Projects Releases Wiki Activity

Added CUDA version warning and fixed cuda_install for 9.2/10.2.

Browse Source
This commit is contained in:
Tim Dettmers 2022-08-09 20:02:47 -07:00
parent f9cbe2fe99
commit 1ced47c504
4 changed files with 11 additions and 3 deletions

8
bitsandbytes/cuda_setup/main.py
View File

@ -45,6 +45,9 @@ def get_cuda_version(cuda, cudart_path):
major = version//1000 major = version//1000
minor = (version-(major*1000))//10 minor = (version-(major*1000))//10
if major < 11:
print('CUDA SETUP: CUDA version lower than 11 are currenlty not supported!')
return f'{major}{minor}' return f'{major}{minor}'
@ -110,6 +113,10 @@ def get_compute_capability(cuda):
def evaluate_cuda_setup(): 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" binary_name = "libbitsandbytes_cpu.so"
cudart_path = determine_cuda_runtime_lib_path() cudart_path = determine_cuda_runtime_lib_path()
if cudart_path is None: if cudart_path is None:
@ -121,6 +128,7 @@ def evaluate_cuda_setup():
print(f"CUDA SETUP: CUDA path found: {cudart_path}") print(f"CUDA SETUP: CUDA path found: {cudart_path}")
cuda = get_cuda_lib_handle() cuda = get_cuda_lib_handle()
cc = get_compute_capability(cuda) 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) cuda_version_string = get_cuda_version(cuda, cudart_path)

2
cuda_install.sh
View File

@ -65,7 +65,7 @@ if [[ -n "$CUDA_VERSION" ]]; then
echo $URL echo $URL
echo $FILE echo $FILE
wget $URL 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 LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$BASE_PATH/$FOLDER/lib64/" >> ~/.bashrc
echo "export PATH=$PATH:$BASE_PATH/$FOLDER/bin/" >> ~/.bashrc echo "export PATH=$PATH:$BASE_PATH/$FOLDER/bin/" >> ~/.bashrc
source ~/.bashrc source ~/.bashrc

2
deploy_from_slurm.sh
View File

@ -202,4 +202,4 @@ if [ ! -f "./bitsandbytes/libbitsandbytes_cuda117_nocublaslt.so" ]; then
fi fi
python -m build python -m build
python -m twine upload dist/* --verbose --repository testpypi python -m twine upload dist/* --verbose

2
setup.py
View File

@ -18,7 +18,7 @@ def read(fname):
setup( setup(
name=f"bitsandbytes", name=f"bitsandbytes",
version=f"0.31.7", version=f"0.31.8",
author="Tim Dettmers", author="Tim Dettmers",
author_email="dettmers@cs.washington.edu", author_email="dettmers@cs.washington.edu",
description="8-bit optimizers and matrix multiplication routines.", description="8-bit optimizers and matrix multiplication routines.",