【ubuntu14.04配置caffe】六——安装编译caffe

在ubuntu14.04下配置caffe主要有以下步骤:

  1. NVIDIA驱动以及cuda和cudnn的安装配置
  2. Intel MKL的安装配置
  3. MATLAB的安装配置
  4. OpenCV的安装配置
  5. Anaconda以及pycuda的安装配置
  6. caffe的安装编译



本文介绍caffe的安装编译



注意:在安装前,请确保显卡驱动、cuda、MKL、matlab、opencv、python依赖包都已正确安装和配置。


1.安装编译glog


鉴于某些原因,这里将该资源传到这里方便大家下载:http://download.csdn.net/detail/budf01/9686363

下载后进入下载目录,执行:
tar zxvf glog-0.3.3.tar.gz
chmod a+x glog-0.3.3 -R
cd glog-0.3.3
./configure
sudo make
sudo make install


2.安装依赖项

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

3.下载caffe

建议先fork到自己的github,再git clone自己的caffe-master

或者直接:
git clone git@github.com:BVLC/caffe.git

(有关git的用法就不在这里讲了,不需要用git就直接下载https://github.com/BVLC/caffe/)

下载后,通过以下命令生成Makefile.config:
cd caffe
cp Makefile.config.example Makefile.config


4.配置Makefile.config:

1.启用cudnn
(如果没有安装cudnn或者不支持cudnn请忽略)

去掉第5行前面的#:
USE_CUDNN := 1

2.启用MKL
第46行修改为:
BLAS := mkl

3.启用matlab
第59行修改为:(修改为你自己的matlab路径)
MATLAB_DIR :=/usr/local/MATLAB/R2015b

4.配置python路径
第79行:
PYTHON_LIB :=/usr/local/lib

如果安装了anaconda,可将关于anaconda的内容取消注释,并将ANACONDA_HOME改为自己的安装路径

5.其他
第90行:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/include /usr/include


5.编译

make all -j4           //将4修改为你的cpu内核数,几核就改成几,可加快编译速度
make test -j4
make runtest -j4
make pycaffe -j4
make matcaffe -j4

在make runtest时出现错误:
[  FAILED  ] 6 tests, listed below:
[  FAILED  ] SGDSolverTest/0.TestSnapshotShare, where TypeParam = caffe::CPUDevice<float>
[  FAILED  ] AdaGradSolverTest/0.TestSnapshotShare, where TypeParam = caffe::CPUDevice<float>
[  FAILED  ] NesterovSolverTest/0.TestSnapshotShare, where TypeParam = caffe::CPUDevice<float>
[  FAILED  ] AdaDeltaSolverTest/0.TestSnapshotShare, where TypeParam = caffe::CPUDevice<float>
[  FAILED  ] AdamSolverTest/0.TestSnapshotShare, where TypeParam = caffe::CPUDevice<float>
[  FAILED  ] RMSPropSolverTest/0.TestSnapshotShare, where TypeParam = caffe::CPUDevice<float>

 
网络上有人使用export MKL_CBWR=AUTO,在重新make runtest通过,但是我在尝试时并没有解决。
如果大家有好的解决办法欢迎留言。

6.使用MNIST数据集进行测试 


1.数据准备:
cd $CAFFE_ROOT
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh


2.训练和测试mnist
./examples/mnist/train_lenet.sh




  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值