如何在caffe中更改GPU模式和CPU模式？

Question

我编译了Caffe而没有复制和编辑Makefile.config.example。我试图在caffe中运行代码，它显示了我的CPU模式。我的问题是，在caffe安装之后是否可以对配置文件进行细化？如果有，怎么样？

~/caffe-master$ SOME_COMMANDS BLAH BLAH

CPU mode

如何将运行从 CPU模式 切换到 GPU模式 ，反之亦然？

如果你帮助我，我真的很感激。我是caffe和python的初学者。感谢

Answer 1

你在哪里得到你的咖啡？ official BVLC release默认为GPU构建。

首先，Makefile.config有一对行：

# CPU-only switch (uncomment to build without GPU support).
# CPU_ONLY := 1

您必须取消注释才能获得仅限CPU的版本。

然后每个求解器文件的最后一行是

solver_mode: GPU

您需要将其更改为 CPU 才能执行CPU。

Answer 2

首先，你必须编辑“Makefile.config.example”，它是“ \ Windows \ CommonSettings.props.example“。

在所有事情发生之前，只需在命令末尾使用“-gpu all”运行你的caffe，就像：

>caffe train lenet_solver.prototxt -gpu all

然后如果再次使用cpu完成，那么我认为您需要使用所提到的文件中的所需版本重建您的caffe。当然在新版本的caffe中，“。props”文件中的默认模式有GPU选项，但必须在系统变量中设置CUDA_PATH（在环境变量中），如下所示：

CUDA_PATH = C：\ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v7.5

虽然如果你的cuda安装正确，它应该自动设置。