会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
happy_fan
博客园
首页
新随笔
联系
订阅
管理
pytorch单机多卡训练
pytorch单机多卡训练
训练
只需要在model定义处增加下面一行:
model = model.to(device) # device为0号 model = torch.nn.DataParallel(model)
载入模型
如果是多GPU载入,没有问题
如果训练时是多GPU,但是测试时是单GPU,会出现报错
解决办法
posted @
2020-06-24 15:18
happy_fan
阅读(
2204
) 评论(
0
)
编辑
收藏
举报
会员力量,点亮园子希望
刷新页面
返回顶部
公告