Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

handle torch cuda crashing in setup

Browse Source
This commit is contained in:
Vladimir Mandic
2023-04-23 11:09:36 -04:00
parent d105074598
commit d64bb50235
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions-builtin/a1111-sd-webui-lycoris
+1 -1
Submodule extensions-builtin/a1111-sd-webui-lycoris updated: f39bc5598b...95149e4073
modules/images.py
+1
View File
@@ -338,6 +338,7 @@ class FilenameGenerator:
'sampler': lambda self: self.p and sanitize_filename_part(self.p.sampler_name, replace_spaces=False),
'model_hash': lambda self: getattr(self.p, "sd_model_hash", shared.sd_model.sd_model_hash),
'model_name': lambda self: sanitize_filename_part(shared.sd_model.sd_checkpoint_info.model_name, replace_spaces=False),
'model_shortname': lambda self: sanitize_filename_part(shared.sd_model.sd_checkpoint_info.name_for_extra, replace_spaces=False),
'date': lambda self: datetime.datetime.now().strftime('%Y-%m-%d'),
'datetime': lambda self, *args: self.datetime(*args), # accepts formats: [datetime], [datetime<Format>], [datetime<Format><Time Zone>]
'job_timestamp': lambda self: getattr(self.p, "job_timestamp", shared.state.job_timestamp),
setup.py
+4 -3
View File
@@ -200,15 +200,16 @@ def check_torch():
log.warning("Torch repoorts CUDA not available")
else:
if torch.version.cuda:
log.info(f'Torch backend: nVidia CUDA {torch.version.cuda} cuDNN {torch.backends.cudnn.version()}')
log.info(f'Torch backend: nVidia CUDA {torch.version.cuda} cuDNN {torch.backends.cudnn.version() if torch.backends.cudnn.is_available() else "N/A"}')
elif torch.version.hip:
log.info(f'Torch backend: AMD ROCm HIP {torch.version.hip}')
else:
log.warning('Unknown Torch backend')
for device in [torch.cuda.device(i) for i in range(torch.cuda.device_count())]:
log.info(f'Torch detected GPU: {torch.cuda.get_device_name(device)} VRAM {round(torch.cuda.get_device_properties(device).total_memory / 1024 / 1024)} Arch {torch.cuda.get_device_capability(device)} Cores {torch.cuda.get_device_properties(device).multi_processor_count}')
except:
pass
except Exception as e:
log.error(f'Could not load torch: {e}')
exit(1)
try:
if 'xformers' in xformers_package:
install(f'--no-deps {xformers_package}', ignore=True)