如何在 caffe 中切换 GPU 模式和 CPU 模式？

Question

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

~/caffe-master$ SOME_COMMANDS BLAH BLAH

CPU mode

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

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

Answer 1

你的 Caffe 从哪里来的？ official BVLC release 默认作为 GPU 构建。

首先，Makefile.config 有两行：

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

您必须取消注释才能获得仅 CPU 的构建。

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

solver_mode: GPU

您需要将其更改为 CPU 以获取 CPU 执行。