在Qt控制台应用程序中使用Caffe

Question

我正在尝试在Qt IDE中使用caffe，

我有以下代码：

#include <QCoreApplication>
#include "caffe/caffe.hpp"

using namespace caffe;

int main(int argc, char *argv[])
{
    QCoreApplication a(argc, argv);
    Caffe::set_mode(Caffe::CPU);
    return a.exec();
}

我只是想在CPU模式下设置Caffe。

我有以下.pro文件

QT += core
QT -= gui

CONFIG += c++11
TARGET = CaffeTest_v0
CONFIG += console
CONFIG -= app_bundle

TEMPLATE = app

INCLUDEPATH += /home/ahmet/caffe/include \
               /home/ahmet/caffe/distribute/include \
               /home/ahmet/caffe/build/lib

 LIBS += -lcaffe

 SOURCES += main.cpp

我有以下问题：

我的问题是我想在CPU模式下设置caffe，我在CPU模式下安装了caffe。

如何在主模式下在CPU模式下设置Caffe？

P.S。我相信我已经安装了caffe，因为我在gnome终端

中通过以下命令进行了测试

＆＃34; caffe train -solver lenet_train_solver.prototxt＆＃34;

我还搜索了以下网站：

http://tzutalin.blogspot.co.uk/2015/05/caffe-on-ubuntu-eclipse-cc.html

Error while including Caffe in C++ Project using cmake

https://github.com/BVLC/caffe/issues/3317

https://groups.google.com/forum/#!topic/caffe-users/DAv5EcsvbIU

Answer 1

在.pro文件中，我应该添加

DEFINES += CPU_ONLY

解决了这个问题。

最终的.pro文件如下：

QT += core
QT -= gui

CONFIG += c++11

TARGET = CaffeTest_v0
CONFIG += console
CONFIG -= app_bundle

TEMPLATE = app

DEFINES += CPU_ONLY



INCLUDEPATH += /home/ahmet/caffe/include \
           /home/ahmet/src/caffe/proto \
           /usr/include/boost

LIBS += -L/home/ahmet/caffe/build/lib \
        -lcaffe \
        -L/usr/lib/x86_64-linux-gnu \
        -lboost_system


SOURCES += main.cpp