Faster RCNN training meets out of memory

rbgirshick / py-faster-rcnn

Faster R-CNN (Python implementation) -- see https://github.com/ShaoqingRen/faster_rcnn for the official MATLAB version

Other

8.12k stars 4.11k forks source link

Open haihaoshen opened 8 years ago

haihaoshen commented 8 years ago

Hi,

When I trained my own image database, there is an error happened: ... syncedmem.cpp:58] Check failed: error == cudaSuccess (2 vs. 0) out of memory

I used the command: ./tools/train_faster_rcnn_alt_opt.py --gpu 0 --net_name ZF --weights data/imagenet_models/ZF.v2.caffemodel --imdb my_train

Anyone can help? Thanks.

ricepot100 commented 8 years ago

You need cudnn or GPU too low