Theano内存使用量是否波动是正常的吗?

时间:2015-09-16 16:28:56

标签: theano deep-learning

我开始使用Theano库,因为我寻求编译&使用带有caffe的C ++进行调试(尽管如此,这是一个非常棒的库:))

无论如何,我制作了深度网络(几乎像CNN),带有千层面,我开始学习我的网络。但是,我的nvidia-smi表明内存使用量一直在波动,我对它感到不好。当我使用caffe时没有显示,因此,学习可能会很慢。

我使用多进程模块提前获取数据集,我的队列状态似乎正确,因此加载数据集不适合我的慢速训练。

我使用T.shared在GPU中预先分配内存,并使用给定的变量生成函数。

有什么想法吗?

谢谢! 快乐学习!

1 个答案:

答案 0 :(得分:1)

我找到了allow_gc选项,在我用allow_gc = False关闭选项之后,一切都很顺利。

我的GPU利用率显示更好,现在学习速度更快。