Caffe可以扩展到多个CPU核心吗？

Question

我希望在32核机器上运行Caffe。

1. caffe是否可以扩展到可用的核心数量以最好地利用它们？
2. 虽然有32个核心，但是我可以让caffe仅使用选定数量的核心吗？

Answer 1

通常 caffe 在其源代码中不支持多个CPU /核心，但它使用 BLAS 例程。

因此，您的问题的答案如下：

1. 是的，但只能通过 BLAS 配置，i。即您的 BLAS 版本应该使用多线程支持进行编译（请参阅相关讨论：here或here - 在第二个链接中，您还可以找到 caffe的一些修改本身）。
2. 同样通过 BLAS （如果使用 openmp 支持编译，您可以将OMP_NUM_THREADS定义为所需的值。）

Answer 2

caffe does not, but you can you Intel caffe which is optimized for CPU and supports multi node https://github.com/intel/caffe/wiki/Multinode-guide