Linux下caffe的安装

来源:互联网 发布:删除淘宝上传的身份证 编辑:程序博客网 时间:2024/06/06 12:52

Linux下caffe的安装

linux:

安装git:sudo apt-get install git

下载caffe: git clone https://github.com/BVLC/caffe.git

安装依赖项: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 libatlas-base-dev

sudo apt-get install python-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

sudo apt-get install python-lmdb

sudo apt-get install python-pip

cd caffe

配置文件的修改

cp Makefile.config.example Makefile.config(如果用CPU跑,把这两个文件的cuda相关都注释掉)

修改需编译的选项)(针对ubuntu15.04以及之后的版本,之前的版本不用修改,用于解决无法找到hdf5.h等文件)

在INCLUDE_DIR 后添加 /usr/lib/x86_64-linux-gnu/hdf5/serial/include

LIBRARY_DIR /usr/lib/x86_64-linux-gnu/gnu/hdf5/serial

去掉CPU—only选项前的注释,使之生效,若需Python layer,则还需去掉WITH_PYTHON_LAYER :=1前的注释

如需编译MATLAB接口,则需修改MATLAB_DIR,改成当前系统下的MATLAB路径

编译:

make all -j4

make test -j4

make runtest

python 接口

sudo pip install -r python/requirements.txt

make pycaffe

matlab 接口

make matcaffe

gedit ~/.bashrc

最后面添加:export PYTHONPATH=/home/francis/caffe/python

rm *~ 可考虑删除因修改而自动生成的备份文件

以上可成功配置好cpu_only

安装GPU版本

应先装好CUDA,条件允许可以可加入cudnn,用于加速卷积运算

CUDA安装步骤待后续完善

注释掉CPU-only,其他与前面相同


转自薛学通的github :https://github.com/xuexuetong1993/xuexuetong1993.github.io


原创粉丝点击