''tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor

Posted 2020-11-19 图米大侠

这是tensorflow 一个经常性错误，错误的原因在于：显卡内存不够。

解决方法就是降低显卡的使用内存，途径有以下几种措施:

1 减少Batch 的大小

2 分析错误的位置，在哪一层出现显卡不够，比如在全连接层出现的，则降低全连接层的维度，把2048改成1042啥的

3 增加pool 层，降低整个网络的维度。

4 修改输入图片的大小

 

总结以下：想进一切办法降维，降低网络的维度。