原文:tensorflow报错 tensorflow Resource exhausted: OOM when allocating tensor with shape

在使用tensorflow的object detection时,出现以下报错 tensorflow Resource exhausted: OOM when allocating tensor with shape 可能的解决方法: 减小训练的batch大小 ...

2018-08-28 16:32 0 951 推荐指数:

查看详情

''tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[?]'' 错误分析

这是tensorflow 一个经常性错误,错误的原因在于:显卡内存不够。 解决方法就是降低显卡的使用内存,途径有以下几种措施: 1 减少Batch 的大小 2 分析错误的位置,在哪一层出现显卡不够,比如在全连接层出现的,则降低全连接层的维度,把2048改成1042啥的 3 增加pool 层 ...

Fri Jun 22 04:34:00 CST 2018 0 4872
tensorflow ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[256,256,15,15] and type float on 报错(显存不够报错

出现这个的原因是gpu显存不够导致的,一般是我们设置了程序根据需求增长导致的 这里,我们设置 allow_growth=False 就可以控制显存使用的增长,从而控制显存的使用,而不会程序运行一半报错。 这种情况是建立在我们的显存比较小的情况下使用的策略,如果说我们显存够大则不 ...

Fri Nov 08 02:50:00 CST 2019 0 351
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM