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