print more info

2020-08-23 20:35:09 +08:00 · 2020-08-23 20:35:09 +08:00 · 54b0799c48
commit 54b0799c48
parent 9dfb887c86
1 changed files with 1 additions and 0 deletions
--- 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()