print GPU device at startup

2026-04-23 03:00:31 -04:00 · 2023-07-01 20:47:11 -04:00
parent 5de820f2dc
commit 1cf61feead
1 changed files with 2 additions and 0 deletions
--- a/invokeai/app/services/model_manager_service.py
+++ b/invokeai/app/services/model_manager_service.py
@@ -168,6 +168,8 @@ class ModelManagerService(ModelManagerServiceBase):
        logger.debug(f'config file={config_file}')

        device = torch.device(choose_torch_device())
+        logger.debug(f'GPU device = {device}')
+
        precision = config.precision
        if precision == "auto":
            precision = choose_precision(device)