From f4996978db44102bc30b2e97820ab63d108c234c Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Sat, 15 Jul 2023 10:22:08 -0700 Subject: [PATCH] Added missing check if LD_LIBRARY_PATH exists. #588 --- bitsandbytes/__main__.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/bitsandbytes/__main__.py b/bitsandbytes/__main__.py index a100b29..a291fab 100644 --- a/bitsandbytes/__main__.py +++ b/bitsandbytes/__main__.py @@ -63,15 +63,16 @@ def generate_bug_report_information(): print('') print_header("LD_LIBRARY CUDA PATHS") - lib_path = os.environ['LD_LIBRARY_PATH'].strip() - for path in set(lib_path.split(':')): - try: - if isdir(path): - print_header(f"{path} CUDA PATHS") - paths = find_file_recursive(path, '*cuda*so') - print(paths) - except: - print(f'Could not read LD_LIBRARY_PATH: {path}') + if 'LD_LIBRARY_PATH' in os.environ: + lib_path = os.environ['LD_LIBRARY_PATH'].strip() + for path in set(lib_path.split(':')): + try: + if isdir(path): + print_header(f"{path} CUDA PATHS") + paths = find_file_recursive(path, '*cuda*so') + print(paths) + except: + print(f'Could not read LD_LIBRARY_PATH: {path}') print('')