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

Added missing check if LD_LIBRARY_PATH exists. #588

Browse Source
This commit is contained in:
Tim Dettmers 2023-07-15 10:22:08 -07:00
parent 6ec4f0c374
commit f4996978db
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
bitsandbytes/__main__.py
View File

@ -63,15 +63,16 @@ def generate_bug_report_information():
print('') print('')
print_header("LD_LIBRARY CUDA PATHS") print_header("LD_LIBRARY CUDA PATHS")
lib_path = os.environ['LD_LIBRARY_PATH'].strip() if 'LD_LIBRARY_PATH' in os.environ:
for path in set(lib_path.split(':')): lib_path = os.environ['LD_LIBRARY_PATH'].strip()
try: for path in set(lib_path.split(':')):
if isdir(path): try:
print_header(f"{path} CUDA PATHS") if isdir(path):
paths = find_file_recursive(path, '*cuda*so') print_header(f"{path} CUDA PATHS")
print(paths) paths = find_file_recursive(path, '*cuda*so')
except: print(paths)
print(f'Could not read LD_LIBRARY_PATH: {path}') except:
print(f'Could not read LD_LIBRARY_PATH: {path}')
print('') print('')