在这个教程中,我们将学习如何用DataParallel 来使用多GPU。 通过PyTorch 使用多个GPU 非常简单。你可以将模型放在一个GPU: device = torch.device("cuda:0") ...
確定! 回上一頁