会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
happy_fan
博客园
首页
新随笔
联系
订阅
管理
2020年5月12日
torch.load CUDA ERROR: out of memory
摘要: 在pytorch中load模型,出现CUDA ERROR:out of memory,基本有下面三个原因: CUDA显存不足 解决办法:换成显存更大的显卡 显卡被占用 当存在多个显卡时,pytorch默认使用0号显卡,当0号显卡被占用时,即使1号显卡空闲仍然会报错 解决办法:让其他显卡不可见,此时0
阅读全文
posted @ 2020-05-12 11:49 happy_fan
阅读(2924)
评论(0)
推荐(0)
编辑
公告