BLAS是一个数学函数接口标准,有很多个实现。按照Caffe官方ubuntu的安装文档默认安装的是ATLAS。这个版本的BLAS不能利用多核CPU,我们将其换为OpenBLAS,可以利用多核CPU并行计算,加快Caffe的分类速度。
源代码安装配置OpenBLAS
1.git代码到本地并安装
git clone https://github.com/xianyi/OpenBLAS.git cd OpenBLAS make -j4 make install2. 修改Caffe的Makefile.config
BLAS := open BLAS_INCLUDE := /opt/OpenBLAS/include BLAS_LIB := /opt/OpenBLAS/lib3.caffe重新make
make clean
make pycaffe
make all -j4
make test && runtest
4.export环境变量
接下来就可以使用caffe的多cpu进行计算了export LD_LIBRARY_PATH=/opt/OpenBLAS/lib/:$LD_LIBRARY_PATHexport OPENBLAS_NUM_THREADS=4