2

我尝试对 DIGITS 使用预训练模型 (VGG 16)，但出现此错误。

错误：检查失败：错误 == cudaSuccess (2 vs. 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

我上传deploy.prototxt并成功地进入了 DIGITSVGG_ILSVRC_16_layers.caffemodel并synset_words.txt用我的数据集进行了测试，该数据集有两个类。

4

1 回答 1

2

有时digits-server无法清除内存。如果您使用的是 ubuntu，请尝试使用此命令：

sudo restart nvidia-digits-server

如果这不起作用并且您再次面对相同的情况，则需要减少batch_size

于 2016-10-20T06:12:40.443 回答