caffe安装教程
注意事项
综合比较了windos,centos与ubuntu三个系统,个人觉得在ubuntu系统下安装caffe较好,并且推荐16.04版本的ubuntu(gcc4.7.7以上,支持C++11,才能编译更加畅通无阻),如果是虚拟机安装的Ubuntu系统,建议硬盘设置大一些,避免其他不必要的麻烦。
1.检查系统是否支持GPU,再选择你安装的版本
通常情况下,gpu比cup处理更快,但一些电脑不支持gpu
输入
- lspci | grep -i nvidia
结果 1不输出结果,说明不支持GPU
2.输出结果,则支持GPU
所以本文选择安装CPU版本
2.安装依赖
----------------------------------------------------------------------------------------------------
依次输入 sudo apt-get install libprotobuf-dev
sudo apt-get install libleveldb-dev
sudo apt-get install libsnappy-dev
sudo apt-get install libopencv-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install protobuf-compiler
sudo apt-get install libgflags-dev
sudo apt-get install libgoogle-glog-dev
sudo apt-get install liblmdb-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libboost-all-dev
------------------------------------------------------------------------------------------------------
3.下载caffe
可以直接用Git下载,先安装git : sudo apt-get install git
选择你想要存放的位置,再执行以下命令 :git clone git://github.com/BVLC/caffe.git
4.编译caffe
1.进入caffe目录: cd /caffe
2.对Makefile.config.example文件备份: cp Makefile.config.example Makefile.config
3.编辑文本 : sudo gedit Makefile.config
1.去掉CPU_ONLY前面的#号
2.配置引用文件路径
4.编译
- sudo make all
- sudo make test sudo make runtest
- sudo make runtest
5设置python接口
caffe支持的python版本较低,如果用python3设置接口,可能会出现不能导入的情况!
我的ubuntu版本是16.04 自带python版本是2.7,所以直接开始:
1.安装pip
sudo apt-get install python-pip
2.安装fortran编辑器(gfrotran),因为安装scipy库时需要它
sudo apt-get install gfortran
3.在caffe根目录下,有个python文件夹,文件夹里面有个requirements.txt,里面有需要的依赖库和版本信息
cd ~/caffe/python for req in $(cat requirements.txt); do pip install $req; done
4.安装结束后验证
sudo pip install -r requirements.txt
5.加入环境变量
sudo gedit ~/.bashrc
在文件最后加入
export PYTHONPATH=~/caffe/python:$PYTHONPATH
注意这里的路径是你系统caffe安装de的路径
7.更新配置
sudo ldconfig
8.编译接口
cd ~/caffe/
make pycaffe
9.验证 python接口