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環境變量
export OPENBLAS_NUM_THREADS=4
export LD_LIBRARY_PATH=/opt/OpenBLAS/lib/:$LD_LIBRARY_PATH
接下來就可以使用caffe的多cpu進行計算了