CODEWITHSUNDEEP
deep-learningcaffe
Karthik Hegde
Karthik Hegde

Reputation: 183

Does Caffe Scale up to multiple CPU Cores?

I wish to run Caffe on a 32 core machine.

  1. Does caffe scale up to available number of cores to utilize them the best?
  2. Although there are 32 cores, can I make caffe use only a selected number of cores?

Upvotes: 3

Views: 611

Answers (2)

z-star
z-star

Reputation: 690

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

Upvotes: 0

avtomaton
avtomaton

Reputation: 4894

Generally caffe doesn't support multiple CPUs/cores in its source code, but it uses BLAS routines.

Thus answers to your questions are the following:

  1. Yes, but only through BLAS configuration, i. e. your BLAS version should be compiled with multithreading support (see related discussions: here or here - at the second link you can also find some modifications for caffe itself).
  2. Also through BLAS (if it was compiled with openmp support, you can define OMP_NUM_THREADS to desired value).

Upvotes: 5

Related Questions