budui/raycv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

print more info

Browse Source
This commit is contained in:
budui 2020-08-23 20:35:09 +08:00
parent 9dfb887c86
commit 54b0799c48
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@ -19,6 +19,7 @@ def log_basic_info(logger, config):
logger.info(f"- CUDA version: {torch.version.cuda}") logger.info(f"- CUDA version: {torch.version.cuda}")
logger.info(f"- cuDNN version: {torch.backends.cudnn.version()}") logger.info(f"- cuDNN version: {torch.backends.cudnn.version()}")
logger.info(f"- GPU type: {torch.cuda.get_device_name(0)}") 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: if idist.get_world_size() > 1:
logger.info("Distributed setting:\n") logger.info("Distributed setting:\n")
idist.show_config() idist.show_config()