log env

Signed-off-by: Vladimir Mandic <mandic00@live.com>

CHANGELOG.md

@@ -51,6 +51,7 @@
   - update `nunchaku==1.1.0`  
   - lora switch logic from force-diffusers to allow-native  
   - split `reference.json`  
+  - print system env on startup  
 - **Fixes**
   - extension tab: update checker, date handling, formatting etc., thanks @awsr  
   - controlnet with non-english ui locales  

launch.py

@@ -67,11 +67,10 @@ def get_custom_args():
             del env['PS1']
         installer.log.trace(f'Environment: {installer.print_dict(env)}')
     env = [f'{k}={v}' for k, v in os.environ.items() if k.startswith('SD_')]
-    installer.log.debug(f'Env flags: {env}')
-    ldpreload = os.environ.get('LD_PRELOAD', None)
-    ldpath = os.environ.get('LD_LIBRARY_PATH', None)
-    if ldpreload is not None or ldpath is not None:
-        installer.log.debug(f'Linker flags: preload="{ldpreload}" path="{ldpath}"')
+    ld = [f'{k}={v}' for k, v in os.environ.items() if k.startswith('LD_')]
+    compute = [f'{k}={v}' for k, v in os.environ.items() if 'TORCH' in k or 'CUDA' in k or 'ROCM' in k or 'MIOPEN' in k]
+    installer.log.debug(f'Flags: sd={env} ld={ld} compute={compute}')
+    installer.log.debug(f'Flags ')
     rec('args')