From d5aba8eaf141df3973d8557e32f3e28cf3b38af2 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Mon, 14 Nov 2022 13:40:55 +0530 Subject: [PATCH] Show free/total mem while starting up --- ui/sd_internal/device_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ui/sd_internal/device_manager.py b/ui/sd_internal/device_manager.py index b6df40a7..2a9a2e9f 100644 --- a/ui/sd_internal/device_manager.py +++ b/ui/sd_internal/device_manager.py @@ -58,7 +58,7 @@ def auto_pick_devices(currently_active_devices): mem_free /= float(10**9) mem_total /= float(10**9) device_name = torch.cuda.get_device_name(device) - print(f'{device} detected: {device_name} - Memory: {round(mem_total - mem_free, 2)}Gb / {round(mem_total, 2)}Gb') + print(f'{device} detected: {device_name} - Memory (free/total): {round(mem_free, 2)}Gb / {round(mem_total, 2)}Gb') devices.append({'device': device, 'device_name': device_name, 'mem_free': mem_free}) devices.sort(key=lambda x:x['mem_free'], reverse=True)