Skip to main content

2 posts tagged with "GPU"

View All Tags

· 3 min read
Vishnu Subramanian

understanding multi-GPU training

Fastai makes training deep learning models on multiple GPUs a lot easier. In this blog, let's look at different approaches to train a model using multiple GPUs.

In PyTorch, you can achieve Multi-GPU training using 2 different approaches.

  • nn.DataParallel
  • torch.distributed