调整单节点 PyTorch,以适应分布式深度学习

了解如何使用 PyTorch 对机器学习模型执行分布式训练。

此笔记本遵循建议的开发工作流。 它首先演示如何在单个节点上训练模型,然后演示如何使用 HorovodRunner 调整代码进行分布式训练。

HorovodRunner PyTorch MNIST 示例笔记本

获取笔记本