caffe小显存问题

问题描述：

本人使用硬件1050 2G显卡，在运行faster-rcnn中的./tools/demo.py时，出现以下问题：

Loaded network /mydata/rcnn/py-faster-rcnn/data/faster_rcnn_models/VGG16_faster_rcnn_final.caffemodel
F0318 09:18:46.357918 17490 syncedmem.cpp:56] Check failed: error == cudaSuccess (2 vs. 0)  out of memory
*** Check failure stack trace: ***

百度了一下，是因为内存或者显存不足导致的。

网上有两种解决方法：

1、更换成zf模型进行预测。

./tools/demo.py --net zf --gpu 0

运行速度很快

2、由于2G显存不变，所以要修改载入图片的数量，减少显存的占用率

这个还没有测试