2017-02-17 3 views
1

私はArtistic Styling transfer tutorial in CNTKを経由しています。私は1つのイメージのために行くことができます追加イメージをスタイルするためにGPUのメモリを使い果たすことを避けるためにジュピターノートブックを再起動する必要があります。エラーコードが吹く。私はちょうど最後の2つのコードブロックを画像のフォルダを読み込んで、各画像の新しいスタイル付き画像を作成するforループに入れたいと思っています。私はGPUメモリをダンプするしかないと思う。それが正しければ、どうしたらいいですか?第2の画像を処理する前にCNTK Artistic Styling TransferでGPUメモリをダンプします

RuntimeError: CUDA failure 2: out of memory ; GPU=0 ; hostname=JOHN ;  expr=cudaMalloc((void**) &deviceBufferPtr, sizeof(AllocatedElemType) * numElements) 

答えて

0

あなたは del content_activations del loss import gc gc.collect() これらはとにかく再定義する必要がありますオブジェクトとあなたが次のcontent_activations前に(したがって、GPUのメモリを解放する)、それらをクリーンアップするこの方法ですとloss持っているの呼び出し試みることができます建設された。

+0

私はこの方法を使って、最初の2枚の画像でそれを行いましたが、3枚目でエラーが発生しました。私はこれらの変数を削除することでこれが良くなっているとは思っていません。あるいは、12 GBのvramと4 GBのデスクトップを使っています。いずれにしても、それはvramを投棄していない。私は約500の推論を作成できるようにしたいと考えています。 – user18101

+0

GPU(12 GBのvramも同様)でこれを確認できますが、それは数十程度の画像の後に起こるという点が異なります。私たちは修正に取り組みます。 –

+0

最新のマスターで修正されました。 2017年2月22日に予定されている次のリリース(2.0ベータ12)には、これらの変更が適用されます。私は1.5GBのVramを超えることなく100回の繰り返しを実行することができました。私のテストケースでは、delステートメントと(更新された)レスポンスのようなガベージコレクタへの呼び出しがあります。彼らが必要なのかどうかはわかりません。 –

-1

CNTK以外のフレームワークを使用できますか? http://somatic.io

+0

無関係な答え。 –

関連する問題