你如何在Tensorflow中从GPU内存中释放张量变量?

时间:2017-11-16 08:51:24

标签: tensorflow

在Tensorflow中,session.close()不会从模型变量中释放GPU内存。我正在按顺序构建和运行多个图形,并且在运行多个图形后,我在GPU中出现内存不足异常,即使我在每次运行后关闭会话并重置默认图形。

使用Session.reset('')解决了这个问题,因为它实际上释放了变量。问题是reset()也会关闭所有活动会话。有没有办法释放像reset()这样的变量,但没有关闭所有会话?

0 个答案:

没有答案