英伟达（NVIDIA）训练深度学习模型神器APEX使用指南

你是否苦闷于教研室卡很少，卡显存不大，不少模型无法跑，是否发愁不能用很大的batch size致使loss无法下降。若是你使用的是PyTorch，恭喜你，你彻底可使用APEX从中解脱出来。python APEX是什么 APEX是英伟达开源的，完美支持PyTorch框架，用于改变数据格式来减少模型显存占用的工具。其中最有价值的是amp（Automatic Mixed Precision），将模型的大

>>阅读原文<<