2017-04-05 31 views
1

我正在運行這個tutorial,它正在MNIST上訓練一個簡單的MLP。我運行在CPU上。在訓練循環中監視我的計算機活動時,我可以看到內存使用量穩步增加,直到達到100%,並且所有內容都崩潰(需要重新啓動)。我用自己的腳本遇到了這個問題,並認爲我犯了一個錯誤,但是當我在上面的教程中遇到同樣的問題時,我開始懷疑Theano中的錯誤,並在github上打開issue內存逐漸增加,直到崩潰,同時用Theano訓練簡單的MLP在CPU上

我在Ubuntu 16.04上,並使用Theano的最新版本。

有沒有人在CPU上遇到同樣的問題? 有沒有人有任何線索可能會導致問題?

回答

2

在github的theano-issues上,我的相應post被提出瞭解決方法。我在這裏傳播答案。

顯然,問題來自Theano 0.9.0中的一個bug。由於這個漏洞需要開發者修復,一個簡單的解決方法是恢復到Theano 0.8.2。它可以這樣輕鬆完成:

pip install theano==0.8.2 

Theano 0.9.0應自動卸載並替換爲舊版本。

結果:相同的代碼,沒有更多的內存泄漏!

+0

我在使用Theano 0.9.0的Windows上遇到了類似的症狀,這爲我解決了這個問題。如果您使用的是Anaconda,則可以使用以下級別降級到0.8.2: conda install theano = 0.8.2 – nickandross