caffe在纯CPU模式下,使用多核运行

偶然发现一个问题,caffe在训练时,如果是只在cpu模式下,超级慢,看了一下,原因在于程序只用了一个核进行训练,因此速度超级慢

如何解决这个问题呢?需要在编译caffe的时候指定使用openblas,并且需要在使用时指定使用的核数,具体操作如下


1、首先安装openblas

-
Bash 代码
1 sudo apt-get install -y libopenblas-dev

2、修改caffe目录下的Makefile.config文件,将BLAS: =atlas修改为BLAS: =open

3、再编译caffe,首先make clean,清除之前的编译结果,再依次执行make all -j16, make test -j16, make runtest -j16,编译caffe。-j16是指用16个核并行编译caffe,可以大大加快编译速度。

4、编译完成后,执行训练前,需要export OPENBLAS_NUM_THREADS=4, 即使用4个核进行训练

你可能感兴趣的:(caffe)