如何仅使用 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 计算相比，CPU 上的 Caffe 会非常慢。