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