Multi-GPU Examples - PyTorch
pytorch.org › tutorials › beginnerMulti-GPU Examples — PyTorch Tutorials 1.10.0+cu102 documentation Multi-GPU Examples Data Parallelism is when we split the mini-batch of samples into multiple smaller mini-batches and run the computation for each of the smaller mini-batches in parallel. Data Parallelism is implemented using torch.nn.DataParallel .