如何在CUDA中使用Lasange.layers Theano后端时释放GPU内存

时间:2019-01-21 10:12:00

标签: python gpu theano-cuda

我目前正在使用lasagne theano后端在python中运行深度学习程序。我已将数据分为几组,以便在训练期间可以管理GPU内存。我面临的一个问题是,一些具有相似数据的组在循环运行时运行良好,并实现了稳定的GPU内存使用,而有些在经过几个周期后耗尽了内存。我想知道是否有一种方法可以在遍历历元时释放python中不需要的gpu内存。

0 个答案:

没有答案