Caffe能扩展到多个CPU核心吗？

Question

我想在32台核心机器上运行Caffe。

1. caffe是否扩大到可用的核心数量来最好地利用它们？
2. 虽然有32个核心，但我能不能让caffe只使用一些选定的核心呢？

Answer 1

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

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

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

Answer 2

caffe没有，但是您可以使用适合CPU并支持多节点https://github.com/intel/caffe/wiki/Multinode-guide的Intel caffe。