Ubuntu16.04(CPU)安装caffe

一、判断是否支持GPU

  输入命令lspci | grep -i nvidia,回车后如果没有任何显示内容,说明不支持GPU。

二、更改下载源

  备份原来的下载源,cp /etc/apt/sources.list  /etc/apt/sources_init.list。使用gedit /etc/apt/sources.list 或者 vim /etc/apt/sources.list 打开sources.list ,将里面内容替换为如下:

 1 deb http://mirrors.aliyun.com/ubuntu/ xenial main
 2 deb-src http://mirrors.aliyun.com/ubuntu/ xenial main
 3 deb http://mirrors.aliyun.com/ubuntu/ xenial-updates main
 4 deb-src http://mirrors.aliyun.com/ubuntu/ xenial-updates main
 5 deb http://mirrors.aliyun.com/ubuntu/ xenial universe
 6 deb-src http://mirrors.aliyun.com/ubuntu/ xenial universe
 7 deb http://mirrors.aliyun.com/ubuntu/ xenial-updates universe
 8 deb-src http://mirrors.aliyun.com/ubuntu/ xenial-updates universe
 9 deb http://mirrors.aliyun.com/ubuntu/ xenial-security main
10 deb-src http://mirrors.aliyun.com/ubuntu/ xenial-security main
11 deb http://mirrors.aliyun.com/ubuntu/ xenial-security universe
12 deb-src http://mirrors.aliyun.com/ubuntu/ xenial-security universe

  更新工具源,输入命令: apt-get update ,回车。

  更新工具软件:输入命令: apt-get upgrade ,回车。

  执行完此条命令之后会出现某些包无法下载,根据提示执行命令: apt-get update --fix-missing ,然后重新执行一下上条命令: apt-get upgrade 。

三、安装依赖包

  按照下面的依赖包安装,每个命令前面加上sudo获取权限。

 1 apt-get install libprotobuf-dev 
 2 apt-get install libleveldb-dev
 3 apt-get install libsnappy-dev 
 4 apt-get install libopencv-dev
 5 apt-get install libhdf5-serial-dev
 6 apt-get install protobuf-compiler
 7 apt-get install libgflags-dev
 8 apt-get install libgoogle-glog-dev
 9 apt-get install liblmdb-dev
10 apt-get install libatlas-base-dev

四、安装git 以及用git clone指令拉取远程仓库GitHub上的caffe项目

  分别输入命令apt-get install git 、 git clone git://github.com/BVLC/caffe.git。

       下载完成后,cd caffe进入caffe目录,执行命令:cp Makefile.config.example Makefile.config 来备份Makefile.config.example文件。

  修改新文件Makefile.config中的配置,键入命令: gedit Makefile.config 或者 vim Makefile.config 。

  ①、去掉CPU_ONLY前面的#号注释,使得CPU_ONLY := 1

  ②、在INCLUDE_DIRS和LIBRARY_DIRS后面分别加入 /usr/include/hdf5/serial 和 /usr/lib/x86_64-linux-gnu/hdf5/serial ,保存退出。

1 # Whatever else you find you need goes here.
2 INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
3 LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial

  ③、如果是python3环境下使用caffe,需要取消对PYTHON_INCLUDE的注释(第79行),并对之前的python2对应的PYTHON_INCLUDE进行注释。(如果是python3.6还需要安装sudo apt-get install python3.6-dev

  ④、分别输入命令sudo make all、sudo make test和sudo make runtest,编译并测试。如果出现错误,输入sudo make clean,解决错误后重新编译和测试。

五、编译python3接口

  ①、先安装python3.5,命令为sudo apt-get install python3-pip。

  ②、安装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

  安装结束后,可以执行如下语句验证:

1 sudo pip install -r requirements.txt

  ③、将caffe根目录下的python文件夹加入到环境变量

    打开配置文件bashrc,输入sudo gedit ~/.bashrc命令,在最后一行加入export PYTHONPATH=/home/cff/caffe/python:$PYTHONPATH语句。

    执行更新配置,命令为sudo ldconfig

  ④、编译python接口

    输入命令cd ~/caffe/ ,执行make pycaffe

  ⑤、验证python接口

    进入python3环境,导入import caffe,如果没有出错,安装成功。

       ⑥、建议使用清华镜像下载比较快:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple example

原文:https://blog.csdn.net/u010193446/article/details/53259294

posted @ 2020-07-23 14:42  嗨_放飞梦想  阅读(234)  评论(0编辑  收藏  举报