使用 Caffe 启用多线程

Caffe 可以在多个核心上运行。一种方法是使用 Caffe 启用多线程来使用 OpenBLAS 而不是默认的 ATLAS。为此,你可以按照以下三个步骤操作:

  1. sudo apt-get install -y libopenblas-dev
  2. 在编译 Caffe 之前,编辑 Makefile.config ,用 BLAS := open 替换 BLAS := atlas
  3. 在编译 Caffe 之后,运行 export OPENBLAS_NUM_THREADS=4 将导致 Caffe 使用 4 个核心。