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