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

Question

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

我在Ubuntu 16.04上，并使用Theano的最新版本。

有没有人在CPU上遇到同样的问题？ 有没有人有任何线索可能会导致问题？

Answer 1

在github的theano-issues上，我的相应post被提出了解决方法。我在这里传播答案。

显然，问题来自Theano 0.9.0中的一个bug。由于这个漏洞需要开发者修复，一个简单的解决方法是恢复到Theano 0.8.2。它可以这样轻松完成：

pip install theano==0.8.2 

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

结果：相同的代码，没有更多的内存泄漏！