Chapter 31
distribution training
Torch.cuda.amp vs Nvidia apex
pytorch从1.6版本开始,已经内置了torch.cuda.amp,采用自动混合精度训练就不需要加载第三方NVIDIA的apex库了。
加载中...
目录
Chapter 31
Torch.cuda.amp vs Nvidia apex
pytorch从1.6版本开始,已经内置了torch.cuda.amp,采用自动混合精度训练就不需要加载第三方NVIDIA的apex库了。