如何避免TensorFlow训练时内存持续膨胀导致进程被强制终止？

本文共计694个文字，预计阅读时间需要3分钟。

在TensorFlow训练过程中，若遇到内存不断增长，最终导致内存不足、进程被杀的情况。在此，我仅记录一下我在项目中遇到这一问题，并未深入探讨造成这一现象的所有原因。以下是我遇到的问题描述：

TensorFlow训练时，遇到内存不断增长，最终导致内存不足，进程被杀死。

在这里我不准备对造成这一现象的所有原因进行探讨，只是记录一下我在项目中遇到的这一问题，下面将对我遇到的内存不断增长的原因进行分析。

在TensorFlow中构造图，是将一些op作为节点加入图中，在run之前，是需要构造好一个图的，所以在run的时候，如果run图中不存在的节点，TensorFlow就会将节点加入图中，随着不断的迭代，造成内存不断的增长，从而导致内存不足。