在CPU上运行快速RCNN时出错:无法使用GPU

时间:2018-07-09 07:21:27

标签: cpu caffe

我正在尝试在CPU上运行Fast RCNN。我正在使用克隆的代码,如下所示: git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

运行py-faster-rcnn / tools / demo.py时,出现以下错误:

警告:在将InitGoogleLogging()写入STDERR之前进行记录 F0625 01:37:25.908700 24397 common.cpp:66]无法在仅CPU的Caffe:检查模式下使用GPU。 *检查失败堆栈跟踪:* 中止

我修改了Makefile以使用CPU。设置了CPU_ONLY:= 1。 我在这里想念或做错了什么?

1 个答案:

答案 0 :(得分:1)

您是否将--cpu标志传递给命令行本身,即您是否按以下方式运行demo.py: py-faster-rcnn / tools / demo.py --cpu?

如果不是这样,请与该标志一起运行。此错误应通过此方法解决。