Multi GPU - Githubissues

CJHMPower / Simultaneous-Traffic-Sign-Detection-and-Classification-with-RetinaNet

72 stars 28 forks source link

Closed zvadaszi closed 5 years ago

zvadaszi commented 6 years ago

Is multi GPU training supported? Thanks

CJHMPower commented 6 years ago

Yes, it is supported, you can enable it by this line net = torch.nn.DataParallel(net, device_ids=range(torch.cuda.device_count()))