使用 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 個核心。