内存逐渐增加,直到崩溃,同时使用Theano在CPU上训练简单的MLP

时间:2017-04-05 14:31:11

标签: python memory machine-learning theano

我正在运行这个tutorial,正在训练一个关于MNIST的简单MLP。我在CPU上运行。在训练循环中监视我的计算机活动时,我可以看到内存使用量正在稳步增加,直到达到100%并且一切崩溃(需要重启)。我用我自己的脚本遇到了这个问题并认为我犯了一个错误,但是当我在上面的教程中遇到同样的问题时,我开始怀疑Theano中的一个错误,并在github上打开了一个issue

我使用的是Ubuntu 16.04并使用Theano的最新版本。

是否有人在CPU上遇到同样的问题? 有没有人知道哪个操作可能导致问题?

1 个答案:

答案 0 :(得分:2)

我已经向github上关于theano-issues的相应post提出了变通方法。我在这里传播答案。

显然,问题来自Theano 0.9.0中的一个错误。由于这种泄漏需要由开发人员修复,一个简单的解决方法是恢复到Theano 0.8.2。它可以很容易地完成:

pip install theano==0.8.2

Theano 0.9.0应自动卸载并替换为旧版本。

结果:相同的代码,没有更多的内存泄漏!