使用 Theano 在 CPU 上训练简单的 MLP 时内存逐渐增加直到崩溃
Memory gradually increasing until crash while training simple MLP on CPU with Theano
我是 运行 这个 tutorial that is training a simple MLP on MNIST. I run on CPU. While monitoring my computer activity in the training loop, I can see that the memory usage is steadily increasing, until it reaches 100% and everything crashes (need to reboot). I was experiencing this problem with my own script and thought I made a mistake, but when I experienced the same issue on the tutorial above, I started suspecting a bug in Theano and also opened an issue 在 github。
我正在使用 Ubuntu 16.04 并使用 Theano 的前沿版本。
有人在 CPU 上遇到同样的问题吗？
有人知道哪个操作可能导致问题吗？
针对 github 上的 theano 问题，我的相应 post 提出了解决方法。我在这里传播答案。
显然，问题来自 Theano 0.9.0 中的一个错误。由于此漏洞需要由开发人员修复，一个简单的解决方法是恢复到 Theano 0.8.2。可以像这样轻松完成：
pip install theano==0.8.2
Theano 0.9.0 应该会自动卸载并替换为旧版本。
结果：相同的代码，不再有内存泄漏！
我是 运行 这个 tutorial that is training a simple MLP on MNIST. I run on CPU. While monitoring my computer activity in the training loop, I can see that the memory usage is steadily increasing, until it reaches 100% and everything crashes (need to reboot). I was experiencing this problem with my own script and thought I made a mistake, but when I experienced the same issue on the tutorial above, I started suspecting a bug in Theano and also opened an issue 在 github。
我正在使用 Ubuntu 16.04 并使用 Theano 的前沿版本。
有人在 CPU 上遇到同样的问题吗？ 有人知道哪个操作可能导致问题吗？
针对 github 上的 theano 问题，我的相应 post 提出了解决方法。我在这里传播答案。
显然，问题来自 Theano 0.9.0 中的一个错误。由于此漏洞需要由开发人员修复，一个简单的解决方法是恢复到 Theano 0.8.2。可以像这样轻松完成：
pip install theano==0.8.2
Theano 0.9.0 应该会自动卸载并替换为旧版本。
结果：相同的代码，不再有内存泄漏！