2017-04-05 7 views
1

私はMNISTで単純なMLPを訓練しているこのtutorialを実行しています。私はCPU上で動作します。トレーニングループでコンピュータのアクティビティを監視しているうちに、メモリ使用量が着実に増加していて、100%に達してすべてがクラッシュする(再起動する必要がある)ことがわかりました。私は自分自身のスクリプトでこの問題を経験していましたが、間違いをしたと思っていましたが、上のチュートリアルで同じ問題を経験したとき、私はTheanoのバグを疑い始め、issueをgithubで開きました。TheanoでCPU上で単純なMLPを訓練している間にクラッシュするまでメモリは徐々に増加する

私はUbuntu 16.04で、Theanoの出血エッジバージョンを使用しています。

CPUで同じ問題が発生している人はいますか? 問題の原因となっているのは誰ですか?

答えて

2

githubのtheano-issuesに関する対応するpostに回避策が提案されています。私はここに答えを広げています。

明らかに、この問題はTheano 0.9.0のバグに由来します。このリークは開発者が修正する必要があるため、簡単な回避策はTheano 0.8.2に戻すことです。

pip install theano==0.8.2 

Theano 0.9.0は​​、自動的にアンインストールされ、古いバージョンに置き換えられる必要があります。

結果:同じコード、それ以上のメモリリークはありません!

+0

私はWindowsでTheano 0.9.0と同様の症状が発生していましたが、これは私のために修正されました。 Anacondaを使用している場合は、次の方法を使用して0.8.2にダウングレードできます。 conda install theano = 0.8.2 – nickandross

関連する問題