会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
不要叫我靓仔了
博客园
首页
新随笔
联系
订阅
管理
2022年11月19日
pytorch 使用单个GPU与多个GPU进行训练与测试
摘要: pytorch 使用单个GPU与多个GPU进行训练与测试 使用多个GPU的原理就是通过上面这句代码将model在每个GPU上分别保存一份,然后对model的输入tensor进行自动的分割,每个GPU计算tensor的一部分,这样就能实现计算量的平均分配。在每个model计算完成之后,DataPara
阅读全文
posted @ 2022-11-19 16:36 watsono
阅读(206)
评论(0)
推荐(0)
编辑
公告