From 1d4b34c0dd7ca5dfdaeb63e209b257802a978659 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Sun, 11 Sep 2022 10:59:45 +0530 Subject: [PATCH] Print GPU name --- ui/sd_internal/runtime.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py index f0ca6412..afcacb34 100644 --- a/ui/sd_internal/runtime.py +++ b/ui/sd_internal/runtime.py @@ -58,6 +58,7 @@ has_valid_gpu = False force_full_precision = False try: gpu_name = torch.cuda.get_device_name(torch.cuda.current_device()) + print('detected: ', gpu_name) has_valid_gpu = True force_full_precision = ('nvidia' in gpu_name.lower()) and ('1660' in gpu_name or ' 1650' in gpu_name) # otherwise these NVIDIA cards create green images if force_full_precision: