会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
小艾衰
博客园
首页
新随笔
联系
订阅
管理
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False.
If you are running on a CPU-only machine,
please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
gpu训练后的模型在无gpu机器上面载入参数的时候
需要设置torch.load(map_loction=torch.device('cpu'))
posted @
2021-10-27 09:52
小艾衰
阅读(
532
) 评论(
0
)
编辑
收藏
举报
刷新页面
返回顶部
公告