Adapt single node PyTorch to distributed deep learning

Learn how to perform distributed training of machine learning models using PyTorch.

This notebook follows the recommended development workflow. It first shows how to train the model on a single node, and then how to adapt the code using HorovodRunner for distributed training.

HorovodRunner PyTorch MNIST example notebook

Get notebook