使用 Caffe 啟用多執行緒
Caffe 可以在多個核心上執行。一種方法是使用 Caffe 啟用多執行緒來使用 OpenBLAS 而不是預設的 ATLAS。為此,你可以按照以下三個步驟操作:
sudo apt-get install -y libopenblas-dev
- 在編譯 Caffe 之前,編輯
Makefile.config
,用BLAS := open
替換BLAS := atlas
- 在編譯 Caffe 之後,執行
export OPENBLAS_NUM_THREADS=4
將導致 Caffe 使用 4 個核心。