如何清除Tensorflow-Keras GPU内存？

Posted 2021-03-31

我在for循环中拟合了一个模型，但是我收到的错误是GPU的内存已满。我在Anaconda Spyder IDE中使用Keras。我的GPU是华硕GTX 1060 6GB。

我也使用过像K.clear_session()，gc.collect()，tf.reset_default_graph()，del custom_model这样的代码，但它们都没有用。 GPU属性说98％的内存已满：

除了numba.cuda.close()之外什么都没有刷新GPU内存，但是不允许我再次使用我的GPU。清除它的唯一方法是重新启动内核并重新运行我的代码。

我正在寻找任何脚本代码来添加我的代码允许我在for循环中使用我的代码并在每个循环中清除GPU。

答案

我在本文中找到了解决方案：

https://github.com/keras-team/keras/issues/12625

谢谢所有参与的人。