caffe 上OpenBLAS的安裝使用

       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 install
2.修改Caffe的Makefile.config

BLAS := open
BLAS_INCLUDE :=  /opt/OpenBLAS/include
BLAS_LIB := /opt/OpenBLAS/lib
3.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進行計算了

發佈了21 篇原創文章 · 獲贊 19 · 訪問量 10萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章