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



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

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

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

我已经在github上提出了一个变通方法,用于我的相应帖子。我在这里传播答案。

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

pip install theano==0.8.2

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

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

最新更新