F0416 22:15:28.995488 17888 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of m

F0416 22:15:28.995488 17888 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of m

今天我在跑crpn的时候，遇见了如下错误：

I0416 22:15:28.233808 17888 net.cpp:744] Ignoring source layer loss_rpn_tlI0416 22:15:28.233809 17888 net.cpp:744] Ignoring source layer loss_rpn_trI0416 22:15:28.233811 17888 net.cpp:744] Ignoring source layer loss_rpn_brI0416 22:15:28.233814 17888 net.cpp:744] Ignoring source layer loss_rpn_blI0416 22:15:28.233815 17888 net.cpp:744] Ignoring source layer roi-dataI0416 22:15:28.308246 17888 net.cpp:744] Ignoring source layer loss_clsI0416 22:15:28.308260 17888 net.cpp:744] Ignoring source layer loss_bboxim_detect: 1/1000 0.309s 0.001sF0416 22:15:28.995488 17888 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of memory*** Check failure stack trace: ***./experiments/scripts/test.sh: line 48: 17888 Aborted (core dumped) ./tools/test_net.py --gpu 0 --def models/${NET}/test.pt --net ${MODEL} --imdb ${TEST_IMDB} --cfg models/${NET}/config.yml ${EXTRA_ARGS}

c 查资料发现，原来是编译faster-rcnn的时候，忘记用cudnn了，于是到

cd caffe-fast-rcnn

修改把caffe配置文件Makefile.config的cudnn那项的注释解除：

# cuDNN acceleration switch (uncomment to build with cuDNN).USE_CUDNN := 1

make cleanmake -j8 && make pycaffe

参考文献

[1].Check failed: error == cudaSuccess (2 vs. 0) out of memory.

​​Region Proposal Network. https://github.com/xhzdeng/crpn

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。