Linux 下深度学习工具 caffe 安装

在一台系统环境较好的 Linux 机器上可以很容易的安装 caffe,但是如果系统本身很旧,又没有 GPU 的话,安装就太麻烦了,所有都得从头做起,本文档旨在尽可能覆盖安装所要采的坑。

步骤:
、caffe 是主要是 C/C++ 和 Python 编写的。首先,得需要将 gcc,g++ 安装好,通过 yum 安装就可以。

、cuda 的安装,如果机器上配置了 NVIDIA 系列 GPU,则需要安装该驱动,推荐安装该驱动至7.0版本,推荐同时安装 cuDNN。安装可以参考http://blog.csdn.NET/xuanyuansen/article/details/43152311,centos 系统下类似。

、caffe 的 layer 开发以 google 的 protobuf 为格式,需要安装该依赖,推荐安装 2.6.1 版本(gcc版本推荐4.4.7),下载地址为https://github.com/google/protobuf/releases/download/v2.6.1/protobuf-2.6.1.tar.gz,按照 README 文件里的步骤安装即可。

、python 安装,caffe 提供了 python 的接口,可以用 python 进行相关开发,而且可以在 ipython-notebook 里面画出网络结构,非常方便。为了使用这些功能,推荐在目标机器上编译安装 python2.7.10,源码下载地址见 python 官网,为了多用户环境考虑,强烈建议安装目标用户自己的 python,可以在自己用户的 home 下建立 usr 文件夹,解压源码后加如下配置命令:

./configure --enable-shared \
     --prefix=${HOME}/usr/local \
     LDFLAGS=“-Wl,—rpath=${HOME}/usr/local/lib”
make && make install

即需要 python 的 lib 文件可以被其他程序共享,rpath 后面就是自已的路径。

安装好后需要配置目标用户下的 bashrc 文件,加入如下行:
PATH=/home/lianhua/usr/local/bin:$PATH
然后 source bashrc 文件,如果进入 python 版本正确则安装成功,否则检查路径重新编译。

python 安装好后需要安装 pip,在官网上下载 get-pip.py 执行即可,然后安装相应版本的 protobuf,同时安装 caffe 的官方说明安装其他的依赖:
for req in $(cat requirements.txt); do pip install $req; done


五、安装 openblas
从 openblas 官网下 http://www.openblas.net/ 下载安装安装包,安装 README 文件安装就好,make&&make install,可能需要 sudo 权限。


六、yum 依赖安装
参加官方文档 http://caffe.berkeleyvision.org/install_yum.html,需要安装上述所有依赖,首先需要安装和系统相应的最新的 repository 文件,然后执行 sudo yum install epel-release,更新站点从而安装最最新的依赖。


七、OpenCV 安装
前置需要安装较新版本的 cmake,去官网下载安装即可。

opencv 安装有两种方式:

1、下载源码包安装,请参考:
http://www.javieriparraguirre.Net/installing-opencv-debian/(依赖的安装,centos 类似)
http://www.jianshu.com/p/a41da779f763(opencv 安装)
2、利用自动脚本安装:
地址:https://github.com/jayrambhia/Install-OpenCV


八、caffe 安装配置
到此处终于可以进入正题,进行 caffe 的安装,如果上述安装过程顺利的话,caffe 的安装也就没有太大问题了。关键在于正确配置 Makefile.config 文件,需要修改如下配置至相应的路径(参照配置文件的注释,疑惑的地方 google 即可):
1、CUDA_DIR
2、BLAS := open,BLAS_INCLUDE :=,BLAS_LIB :
3、PYTHON_INCLUDE :=
注意此处需要添加 python 下 numpy 的路径
4、PYTHON_LIB :=
5、WITH_PYTHON_LAYER := 1
注意此处需要将 python 的 hdf5 安装路径添加到:INCLUDE_DIRS := $(PYTHON_INCLUDE) 和 LIBRARY_DIRS := $(PYTHON_LIB)

上述配置文件修改好后,回到 caffe 的根目录,执行 make all -j4,注意此处进程不要过多,否则会出现诡异的错误;
make test -j4
make runtest -j4
至此 caffe 主程序编译完毕。


下面编译 pycaffe,至执行
make pycaffe
make distribute
执行完后修改 bashrc 文件,添加
PYTHONPATH=${HOME}/caffe/distribute/python:$PYTHONPATH
LD_LIBRARY_PATH=${HOME}/caffe/build/lib:$LD_LIBRARY_PATH
使得 python 能够找到 caffe 的依赖。
进入 python,import caffe,如果成功则说明一切 ok,否则检查路径从头再来,甚至需要重新编译 python。


转载:http://blog.csdn.net/xuanyuansen/article/details/46786597

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值