Single node PyTorch to distributed deep learning

Note

The managed MLflow integration with <Databricks> on <GCP> requires Databricks Runtime for Machine Learning 9.1 LTS or above.

This notebook follows the recommended development workflow.

HorovodRunner PyTorch MNIST example notebook

Open notebook in new tab