From 54b0799c48f9cd999b7de450d08c493491d8a498 Mon Sep 17 00:00:00 2001
From: budui <wangbudui@foxmail.com>
Date: Sun, 23 Aug 2020 20:35:09 +0800
Subject: [PATCH] print more info

---
 main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/main.py b/main.py
index eeba793..eca18d9 100644
--- a/main.py
+++ b/main.py
@@ -19,6 +19,7 @@ def log_basic_info(logger, config):
     logger.info(f"- CUDA version: {torch.version.cuda}")
     logger.info(f"- cuDNN version: {torch.backends.cudnn.version()}")
     logger.info(f"- GPU type: {torch.cuda.get_device_name(0)}")
+    logger.info(f"- GPU util: \n{torch.cuda.memory_summary(0)}")
     if idist.get_world_size() > 1:
         logger.info("Distributed setting:\n")
         idist.show_config()