2

事前トレーニング済みのモデル (VGG 16) を DIGITS に使用しようとしましたが、このエラーが発生しました。

エラー: チェックに失敗しました: エラー == cudaSuccess (2 対 0) メモリ不足

と

conv2_2 does not need backward computation.
relu2_1 does not need backward computation.
conv2_1 does not need backward computation.
pool1 does not need backward computation.
relu1_2 does not need backward computation.
conv1_2 does not need backward computation.
relu1_1 does not need backward computation.
conv1_1 does not need backward computation.
data does not need backward computation.
This network produces output label
This network produces output softmax
Network initialization done.
Solver scaffolding done.
Finetuning from /home/digits/digits/jobs/20161020-095911-9d01/model.caffemodel
Attempting to upgrade input file specified using deprecated V1LayerParameter: /home/digits/digits/jobs/20161020-095911-9d01/model.caffemodel
Successfully upgraded file specified using deprecated V1LayerParameter
Attempting to upgrade input file specified using deprecated input fields: /home/digits/digits/jobs/20161020-095911-9d01/model.caffemodel
Successfully upgraded file specified using deprecated input fields.
Note that future Caffe releases will only support input layers and not input fields.
Check failed: error == cudaSuccess (2 vs. 0)  out of memory

DIGITSにアップロードdeploy.prototxtしVGG_ILSVRC_16_layers.caffemodelてsynset_words.txt成功し、2 つのクラスを持つデータセットでテストしました。

4

1 に答える 1

2

時々digits-serverメモリをクリアできません。ubuntu を使用している場合は、次のコマンドを試してください。

sudo restart nvidia-digits-server

これが機能せず、再び同じことに直面する場合は、batch_size

于 2016-10-20T06:12:40.443 に答える