
pytorch-distributed:PyTorch分布式训练快速入门及基准测试
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
Distribution是实现所有目标的核心。关键点:笔者开发了一个基于PyTorch的不同加速库在ImageNet上进行演示文稿(单机多卡版),需要的同学可以将所需部分复制到自己的项目中参考(Github链接请见下文)。支持简便且高效的并行计算,通过torch.distributed实现并行训练的加速。 torch.multiprocessing可替代启动器以提高效率。 apex提供了额外的加速功能。 horovod实现了优雅的分布式训练方案,而GPU集群上的分布式训练则更加复杂和高效。补充说明:在ImageNet上使用4块Tesla V100-PICE进行测试发现,Apex的加速效果最佳,但与HorovodDistributed的效果差异不大。平时建议直接使用内置的Distributed功能以获得更好的性能表现。
全部评论 (0)
还没有任何评论哟~


