我正在运行这个tutorial,正在训练一个关于MNIST的简单MLP。我在CPU上运行。在训练循环中监视我的计算机活动时,我可以看到内存使用量正在稳步增加,直到达到100%并且一切崩溃(需要重启)。我用我自己的脚本遇到了这个问题并认为我犯了一个错误,但是当我在上面的教程中遇到同样的问题时,我开始怀疑Theano中的一个错误,并在github上打开了一个issue。
我使用的是Ubuntu 16.04并使用Theano的最新版本。
是否有人在CPU上遇到同样的问题? 有没有人知道哪个操作可能导致问题?
答案 0 :(得分:2)
我已经向github上关于theano-issues的相应post提出了变通方法。我在这里传播答案。
显然,问题来自Theano 0.9.0中的一个错误。由于这种泄漏需要由开发人员修复,一个简单的解决方法是恢复到Theano 0.8.2。它可以很容易地完成:
pip install theano==0.8.2
Theano 0.9.0应自动卸载并替换为旧版本。
结果:相同的代码,没有更多的内存泄漏!