会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
happy_fan
博客园
首页
新随笔
联系
订阅
管理
2020年6月24日
pytorch单机多卡训练
摘要: pytorch单机多卡训练 训练 只需要在model定义处增加下面一行: model = model.to(device) # device为0号 model = torch.nn.DataParallel(model) 载入模型 如果是多GPU载入,没有问题 如果训练时是多GPU,但是测试时是单G
阅读全文
posted @ 2020-06-24 15:18 happy_fan
阅读(2204)
评论(0)
推荐(0)
编辑
公告