Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

timers reporting

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-14 09:18:24 +02:00
parent 5c495c29cd
commit 0ac89385a0
13 changed files with 50 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+4 -1
View File
@@ -143,7 +143,10 @@ def get_gpu_info():
}
elif backend == 'cuda' or backend == 'zluda':
return {
'device': f'{torch.cuda.get_device_name(torch.cuda.current_device())} n={torch.cuda.device_count()} arch={torch.cuda.get_arch_list()[-1]} capability={torch.cuda.get_device_capability(device)}',
'device': f'"{torch.cuda.get_device_name(torch.cuda.current_device())}"',
'count': torch.cuda.device_count(),
'arch': torch.cuda.get_arch_list()[-1],
'capability': torch.cuda.get_device_capability(torch.cuda.current_device()),
'cuda': torch.version.cuda,
'cudnn': torch.backends.cudnn.version(),
'driver': get_driver(),