没有合适的资源?快使用搜索试试~ 我知道了~
pytorch中 gpu与gpu、gpu与cpu 在load时相互转化操作
0 下载量 26 浏览量
2021-01-20
00:02:55
上传
评论 1
收藏 43KB PDF 举报
温馨提示
试读
1页
问题描述 有时在加载已训练好的模型时,会出现 out of memory 的错误提示,但仔细检测使用的GPU卡并没有再用且内存也没有超出。 经查阅发现原来是训练模型时使用的GPU卡和加载时使用的GPU卡不一样导致的。个人感觉,因为pytorch的模型中是会记录有GPU信息的,所以有时使用不同的GPU加载时会报错。 解决方法 gpu之间的相互转换。即,将训练时的gpu卡转换为加载时的gpu卡。 torch.load(‘modelparameters.pth’, map_location={‘cuda:1′:’cuda:0’}) # gpu 1 –> gpu 0 当然,你也可以直接将加载模型时使用
资源推荐
资源评论
资源评论
weixin_38663973
- 粉丝: 2
- 资源: 942
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- 封装swagger组件,提供全新UI以及无状态登录接口调用解决方案
- 小龙坎支局2024年4月渠道积分核对数据.xlam
- onlyoffice搭建及与alist使用的view.html
- Quadcopter-UAV-attitude-estimation-linux常用命令大全demo
- Quadcopter-UAV-attitude-estimation-based-on-数据库课程设计
- pbdlib-python-master.zip
- 43904245495352013_base.apk
- 基于springboot+vue + redis的工作流审批系统
- 【qt基础语法及实际案例解析】qt基础语法及实际案例解析
- 结课作业.psd
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功