在Ubuntu上安装caffe

来源:互联网 发布:淘宝代运营被骗 编辑:程序博客网 时间:2024/06/06 18:24
Caffe是一款优秀的深度学习编程框架,这段时间因为需要使用LSTM、CNN等,因此尝试在有GPU的主机上安装Caffe,系统是ubuntu。

首先从官网下载Caffe的最新版源码包,在安装之前,先需要更新部分系统依赖:
sudo apt-get install libprotobuf-dev libleveldb-devlibsnappy-dev libopencv-dev libhdf5-serial-devprotobuf-compiler
sudo apt-get install --no-install-recommendslibboost-all-dev

在安装Caffe之前,需要安装cuda和Nvidia驱动,由于主机上的GPU有点问题,因此先略过,在makefile.config里面,把CPU_ONLY=1的注释先取消掉,就可以用CPU的模式来处理

然后需要安装BLAS库,这里我们使用的是OpenBlas,而Caffe默认的是Atlas,因此需要修改配置文件,首先cpMakefile.config.example Makefile.config,然后修改这个默认的配置文件:vimMakefile.config
默认选项是BLAS := atlas,我们修改成BLAS:=open即可,另外保险起见,可以把选项下面的路径也设置为OpenBlas的对应路径

然后就是输入make all安装,可以加入-j4之类的,多线程编译加快速度
然后运行make test,再运行make runtest,出现问题ibopenblas.so.0: cannot openshared object file
解决办法是修改profile文件,加入LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/hadoop/whx/OpenBLAS/lib/

然后makeruntest,测试Caffe各部件的运行情况,如果最终全部OK,然后显示PASSED,说明安装完成。
原创粉丝点击