马春杰杰原来是：
|
1 2 3 |
use_cuda = (args.gpu >= 0) and torch.cuda.is_available() device = torch.device("cuda" if use_cuda else "cpu") model = model.to(device) |
这时用的是单GPU，我们需要改为多GPU，需要在model.to(device)之前加一句
|
1 |
model = nn.DataParallel(model, device_ids = [0, 1]) |
即可。
试试
遇到同样的问题
可以
非常需要看到方案
问题一模一样
遇到了问题需要解决
好好学习，天天向上。加油
想看答案呀