Closed miaowang525 closed 2 years ago
Hi @miaowang525 , I did not try to train with multiple GPUs for this project, but it should be straightforward with DataParallel in PyTorch (simply do net = torch.nn.DataParallel(model, device_ids=[0, 1, 2]
), or DistributedDataParallel (recommended by PyTorch, but required more modifications).
Hope it helps.
Best, Songyou
hello Your work is great and i am very interested in your article, and I have some questions to ask you. I have five GPUs, how to call multiple GPUs to run code during training @pengsongyou