如何清除Tensorflow-Keras GPU内存?

问题描述 投票:0回答:1

我在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。

tensorflow keras out-of-memory spyder
1个回答
1
投票

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

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

谢谢所有参与的人。

© www.soinside.com 2019 - 2024. All rights reserved.