原来是:
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 年没有更新,如果文章内容或图片资源失效,请留言反馈,我们会及时处理,谢谢!
这种方法用于docker安装的alist不太友好
不错,感谢
就是回调地址是那个?,你的另一篇文章的回调地址还是用不了!
Thanks for finally writing about > Typecho中的Widget_Options用法-马春杰杰 < Loved it!
大佬,没看懂,回调地址是什么?
感谢您与我们分享此类文章。
感谢分享,文章不错
很方便诶,但下载了以后网页图片的顺序会不会乱啊