会员
众包
新闻
博问
闪存
赞助商
HarmonyOS
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
笨笨和呆呆
博客园
首页
新随笔
联系
订阅
管理
2022年3月17日
pytorch】gpu训练的模型无法在cpu下使用问题解决
摘要: 最近遇到一个问题,发下gpu训练的模型在cpu下不能跑,很难受。结果发现原来是自己使用了DataParallel的问题,我保存的是整个模型。 model=DataParalle(model)torch.save(‘xx.pkl’,model) 然后把模型copy到自己电脑上用cpu跑 model=t
阅读全文
posted @ 2022-03-17 09:34 笨笨和呆呆
阅读(1247)
评论(0)
推荐(0)
公告