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メモリ
GPUを使用している他のプロセスはありますか? – Shai
これが一度しか動作しない場合は、この種のエラーが発生します。このエラーは、メモリが適切に解放されていないことを意味します。それが本当に記憶を超えたら、あなたはいつもそれを得るでしょう。また、バッチサイズ、画像サイズ、カーネルサイズ、出力パラメータサイズ、ネットワークレイヤーの深さなどのためにメモリを超えることができます。 –
'del your_net'コマンドを使用してPythonでネットオブジェクトを解放するとどうなりますか? – Dale