python - 使用 Theano 在 CPU 上训练简单的 MLP 时，内存逐渐增加直至崩溃

Question

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

我使用的是 Ubuntu 16.04，并使用 Theano 的前沿版本。

有人在CPU上遇到同样的问题吗？有人知道哪个操作可能导致问题吗？

Answer 1

已向我对应的 post 提出了解决方法关于 github 上的 theano-issues。我在这里传播答案。

显然，该问题来自 Theano 0.9.0 中的一个错误。由于开发人员需要修复此泄漏，一个简单的解决方法是恢复到 Theano 0.8.2。可以像这样轻松完成:

pip install theano==0.8.2

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

结果:相同的代码，不再有内存泄漏!