小编Sep*_*ari的帖子

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

我还使用过以下代码：K.clear_session()、gc.collect()、tf.reset_default_graph()del custom_model 但它们都不起作用。GPU 属性表明 98% 的内存已满：

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

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