2016-10-17 11 views
3

gpuでpycaffeを使用してニューラルネットワークを実行しようとしています。チェック失敗:エラー== cudaSuccess(2対0)メモリ不足

これは、スクリプトを初めて呼び出すときに機能します。 同じスクリプトを2回目に実行すると、CUDAはそのエラーをタイトルに投げます。

バッチサイズが1で、この時点でのイメージサイズは243x322、gpuは8GB RAMです。

私はメモリをリセットするコマンドがないと思いますか?

ありがとうございました!

編集:私はWindows上でカフェを実行しています:

たぶん私はいくつかのことを明確にすべきです。

私はpython script.pyでスクリプトを呼び出すと、プロセスが終了し、gpuメモリが解放されるので、これは動作します。私はデバッグに使用ipythonで

は、GPUのメモリが実際に解放されません(1つのパスの後、8 BGの6は、使用中のnvidia-SMIの提案に感謝です!)

ので、私が探していますが、私はの線に沿って、pyhton内から呼び出すことができますコマンドです。

実行ネットワーク

プロセスイメージ出力

空きGPUメモリ

+0

GPUを使用している他のプロセスはありますか? – Shai

+0

これが一度しか動作しない場合は、この種のエラーが発生します。このエラーは、メモリが適切に解放されていないことを意味します。それが本当に記憶を超えたら、あなたはいつもそれを得るでしょう。また、バッチサイズ、画像サイズ、カーネルサイズ、出力パラメータサイズ、ネットワークレイヤーの深さなどのためにメモリを超えることができます。 –

+0

'del your_net'コマンドを使用してPythonでネットオブジェクトを解放するとどうなりますか? – Dale

答えて

2

これは、GPUでメモリ不足が発生した場合に発生します。最初のスクリプトを正しく停止してもよろしいですか?システム上で実行中のプロセス(ubuntuのps -A)を確認し、pythonスクリプトがまだ実行中かどうかを確認してください。もしあればそれを殺す。また、あなたのGPU(nvidia-smi)で使用されているメモリをチェックする必要があります。

3

あなたのGPUメモリは解放されていません。これは、前のプロセスが停止しても終了しない場合に発生します。私の答えhereを参照してください。

関連する問題