查看: 2603|回复: 0

[远程开发] OOM/out of memory解决方法(显存不足)

[复制链接]

166

主题

616

帖子

1万

积分

xdtech

Rank: 5Rank: 5

积分
11584
发表于 2019-4-16 14:22:59 | 显示全部楼层 |阅读模式
在AI开发中,如果出现"OOM..."或者"cuda out of memory"等字样,很有可能是显存不足导致的。
解决方法:
1)降低batchsize,比如将batchsize从64降至32,从32降至16等等
2)更换大显存的显卡,比如将1050ti(显存4G)、1060(显存6G)更换为1080ti(显存11G)
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表