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