Happy清子 发表于 2019-4-16 14:22:59

OOM/out of memory解决方法（显存不足）

在AI开发中，如果出现"OOM..."或者"cuda out of memory"等字样，很有可能是显存不足导致的。
解决方法：
1）降低batchsize，比如将batchsize从64降至32,从32降至16等等
2）更换大显存的显卡，比如将1050ti(显存4G)、1060(显存6G)更换为1080ti（显存11G）
页: [1]
查看完整版本: OOM/out of memory解决方法（显存不足）