除去网上介绍的那些方法,如测试一个合适的num_worker,或者设置pin_memory为true以外,还有一些方法,比如这里有几个工具是专门用来测试GPU的使用情况的,如pytorch下的工具包
TORCH.UTILS.BOTTLENECK
还有英伟达官方提供的一个工具包:
PyProf
除此之外,pytorch的另一个工具:
PyTorch Profiler
也可以来做这个事情
此外,这里有一个别的更强的工具,我感觉说不定是破局的方法,
感谢大佬提供的解决方案
除去网上介绍的那些方法,如测试一个合适的num_worker,或者设置pin_memory为true以外,还有一些方法,比如这里有几个工具是专门用来测试GPU的使用情况的,如pytorch下的工具包
TORCH.UTILS.BOTTLENECK
还有英伟达官方提供的一个工具包:
PyProf
除此之外,pytorch的另一个工具:
PyTorch Profiler
也可以来做这个事情
此外,这里有一个别的更强的工具,我感觉说不定是破局的方法,
感谢大佬提供的解决方案