马春杰杰原来是：
|
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]) |
即可。
本文最后更新于2019年12月25日，已超过 1 年没有更新，如果文章内容或图片资源失效，请留言反馈，我们会及时处理，谢谢！
11 2个月前 (06-09)说：
小笨蛋 3个月前 (04-27)说：
123 4个月前 (03-20)说：
查克拉 4个月前 (03-19)说：
mumujun12345 4个月前 (03-18)说：
感谢分享啊 4个月前 (03-16)说：
langjitianya 5个月前 (03-04)说：
骏 6个月前 (01-28)说：
