系统环境
ubuntu14.04
说明
本来我配置的是GPU环境下的框架的,参考的是我师兄的博客,链接为:https://blog.youkuaiyun.com/PrincePaul3/article/details/52094886点击打开链接
但是在配置完成后,在运行demo时,有时候能出结果,而大部分情况下都会报错,错误如下:
Check failed: error == cudaSuccess (2 vs. 0) out of memory
*** Check failure stack trace: ***
已放弃 (核心已转储)
显然这是我的GPU显存不够,因此我只能转为在CPU环境下运行我的demo。因为我之前配置gpu环境时,前期的编译caffe,以及编译faster-cnn已经完成,具体参考上面的链接。下面我只需要修改py-faster-rcnn相关操作,使其可以在CPU环境下运行即可。
修改步骤:
(1)将py-faster-rcnn/lib/fast_rcnn/config.py的__C.USE_GPU_NMS =