---
**Source**: https://github.com/unionai/unionai-docs/blob/main/content/tutorials/model-training/mnist-classifier/pytorch-single-node-multi-gpu.md
**HTML**: https://www.union.ai/docs/v1/flyte/tutorials/model-training/mnist-classifier/pytorch-single-node-multi-gpu/