python - Keras 内存泄漏

Question

我正在使用 Keras TensorFlow 1.8 并且在我的 gpu (1080 ti) 中有内存泄漏。训练网络后，即使完全关闭 python，我的内存也会被使用。在 nvidia-smi 中，它不再显示 python，但内存使用情况仍然存在。

我无法重新启动计算机，因为其他用户正在运行进程（我确定他们没有使用 gpu）。

[编辑：我上传了错误的截图]

标签： pythontensorflowmemory-leakskerasgpu

Answer 1

总是

K.clear_session()

其中 K 定义为

from keras import backend as K

在您的处理结束时。

它可以防止 TensorFlow 内存泄漏。

你也可以试试

import gc
gc.collect()

或者 ，

从您的 tf 会话开始，使用整个 gpu 功能防止 tensorflow：

import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)