如何仅使用CPU运行项目Caffe

Question

我在没有GPU的情况下使用Ubuntu 14.04我希望运行此代码（with CPU only）。我想用CPU（而不是GPU）运行此代码：https://github.com/smallcorgi/Faster-RCNN_TF。我该怎么办？

Answer 1

您所指的Github存储库是Faster-RCNN的 TensorFlow 实现，而不是 Caffe 。

如果您想使用 Caffe 实施，则必须使用此存储库：https://github.com/rbgirshick/py-faster-rcnn

您必须编辑用于训练和测试模型的Python脚本，例如train_faster_rcnn_alt_opt.py，以便caffe.set_mode_gpu()代替caffe.set_mode_cpu()。您可能需要通过编辑Makefile.config文件来编译 Caffe 以删除CUDNN和CUDA支持。

请注意，与GPU计算相比，Caffe在CPU上非常慢。