如何運行一個項目與CPU只Caffe

Question

我使用Ubuntu 14.04沒有GPU，我想運行此代碼（with CPU only）。我想用CPU（而不是GPU）運行此代碼：https://github.com/smallcorgi/Faster-RCNN_TF。我該怎麼辦 ？

Answer 1

你指的是GitHub的倉庫是一個TensorFlow實現更快RCNN的，不來自Caffe。

如果你想使用來自Caffe執行，你必須使用這個倉庫：https://github.com/rbgirshick/py-faster-rcnn

你必須編輯用於訓練和測試模型，例如在Python腳本train_faster_rcnn_alt_opt.py，以便將caffe.set_mode_gpu()行替換爲caffe.set_mode_cpu()。您可能需要編輯Caffe，方法是編輯Makefile.config文件以刪除CUDNN和CUDA支持。

請注意，與GPU計算相比，CPU上的Caffe將爲非常慢。