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

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

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


1、首先安装openblas

-
Bash 代码
1sudo 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个核进行训练
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值