您的位置:首页 > Web前端

Caffe安装教程:Ubuntu16.04(CPU)

2018-03-18 13:54 871 查看

一、检查自己电脑是否具有GPU

lspci | grep -i nvidia

如果未显示任何内容,则证明你的电脑不支持GPU。
二.安装必要的环境
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  
sudo apt-get install libatlas-base-dev

三.下载caffe

sudo apt-get install git  安装Git
git clone git://github.com/BVLC/caffe.git  下载caffe,就可以在根目录下看到有个caffe目录了

四.编译caffe

(1)进入caffe目录
cd caffe/
(2)生成Makefile.config文件,这里是将caffe目录下自带的Makefile.config.example文件复制一份并更名为Makefile.config,命令如下:
cp Makefile.config.example Makefile.config

(3)修改Makefile.config文件中的配置
1  sudo gedit Makefile.config
2  去掉CPU_ONLY前面的#号,使得CPU_ONLY := 1
3  配置引用文件路径(主要是HDF5的路径问题):
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial
BLAS := atlas

(4)编译 Caffe(这个make默认是用CPU单核运算,如果想要快一点, 比如我想使用四核, 在make后面加上-j4标签。

如果上面4行某一行报错之后想要重试,建议先make clean再重新开始。)
sudo make all
sudo make test

sudo make runtest 

(执行玩runtest后,会有结果显示:显示1098个用例执行成功,结果PASSED!,如果执行结果失败,则需要执行语句sudo make clean,然后解决掉问题重新编译

sudo make clean #清除原编译

sudo make -j4  # make -j4或者make -j8 这根据你自己电脑来定

sudo make pycaffe   #最关键的一个

sudo  make runtest   #测试编译)

五.编译python接口
(1)安装pip
sudo apt-get install python-pip
sudo apt-get install python-numpy

(2)安装Python接口依赖库(在caffe根目录下,有个python文件夹,文件夹里面有个requirements.txt,里面有需要的依赖库和版本信息,按照其安装即可,在安装前,需要先安装fortran编辑器(gfrotran),因为安装scipy库时需要它,命令如下:)
1 sudo apt-get install gfortran
2 cd ~/caffe/python
3 for req in $(cat requirements.txt); do pip install $req; done

安装结束后,可以执行如下语句验证:
sudo pip install -r caffe/python/requirements.txt
(可以看到,安装成功的,都会显示Requirement already satisfied,没安装成功的,会继续进行安装。)

(3)将caffe根目录下的python文件夹加入到环境变量
先打开配置文件bashrc
sudo gedit ~/.bashrc

在文件的最后面添加
export PYTHONPATH=/home/c/caffe/python:$PYTHONPATH
注意,这里路径根据每个人电脑则不一样

然后执行更新配置(有时更新不起作用时,可以重启电脑就会有效):
sudo ldconfig

编译python接口
cd ~/caffe/
make pycaffe

验证python接口
进行python环境,引入caffe包,如果没有报错则安装成功!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: