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

Question

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

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

Answer 1

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

因此,您的问题的答案如下:

1. 是的,但只能通过BLAS配置,即您的BLAS版本应该使用多线程支持进行编译(参见相关讨论:此处或此处 - 在第二个链接中,您还可以找到对caffe本身的一些修改).
2. 同样通过BLAS(如果使用openmp支持编译,则可以定义OMP_NUM_THREADS为所需的值).