[mcj]Pytorch单GPU改为多GPU训练

发表日期：2019-12-23 马春杰杰分类：Pytorch学习阅读(21) 评论(0) 本文内容有更新，最后修改日期为：2019年12月25日

原来是：

use_cuda = (args.gpu >= 0) and torch.cuda.is_available()
device = torch.device("cuda" if use_cuda else "cpu")
model = model.to(device)

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)之前加一句

model = nn.DataParallel(model, device_ids = [0, 1])

1	model = nn.DataParallel(model, device_ids = [0, 1])

即可。

如果你对这篇文章有什么疑问或建议，欢迎下面留言提出，我看到会立刻回复！

打赏

标签：pytorch

好好学习，天天向上。加油！！！

马春杰杰博客始建于2017年6月14日，博客主要分享网站建设中遇到的问题及解决方案、自己在读书过程中的心得体会及分享一些自觉有意义的音视频内容，记录些生活中的琐事，希望博客能督促怠惰的自己不断学习，不断进步。

不公开

※ 微博 ※

不公开

※ 支付宝好友 ※

不公开

※ 个人QQ号 ※

总金额：1.00 马杜币