caffe 上OpenBLAS的安装使用

来源:互联网 发布:linux 发送at指令 编辑:程序博客网 时间:2024/05/01 23:13

       BLAS是一个数学函数接口标准,有很多个实现。按照Caffe官方ubuntu的安装文档默认安装的是ATLAS。这个版本的BLAS不能利用多核CPU,我们将其换为OpenBLAS,可以利用多核CPU并行计算,加快Caffe的分类速度。

源代码安装配置OpenBLAS

1.git代码到本地并安装

git clone https://github.com/xianyi/OpenBLAS.gitcd OpenBLASmake -j4make install
2.修改Caffe的Makefile.config

BLAS := openBLAS_INCLUDE :=  /opt/OpenBLAS/includeBLAS_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进行计算了

0 0
原创粉丝点击