wyf425的博客

原创 训练深度学习模型时报错**ResourceExhaustedError:** OOM when allocating tensor with shape原因和解决方法分析

训练深度学习模型时报错ResourceExhaustedError: OOM when allocating tensor with shape原因和解决方法分析一、报错内容报错内容如下图所示：这是在采用tensorflow训练网络时非常常见的报错形式。二、原因分析**核心原因：**GPU显存不足。表面原因：Batchsize太大；图片尺寸太大；池化层池化效果不佳；等等。二、解决方法**核心方法：**使用一切手段减少模型内存占用或扩大显卡内存或采用其他内存。开源节流相辅相成.