Ubuntu虚拟机下caffe安装配置

Ubuntu虚拟机下caffe安装配置

前些日子搭建caffe环境，CSDN总是不那么靠谱，花了些时间装好了，写这篇博客记录一下，希望能帮助其他人一步到位，少踩坑。

首先有个注意的地方：
虚拟机是没法用显卡（GPU）的，所以使用虚拟机安装 caffe 的小伙伴就免于折腾显卡驱动之类的东西了，我开始没注意，花了很多时间倒腾 CUDA 失败，后面才反应过来，这TM是虚拟机环境！！！

环境：ubuntu16.04 、python2.7

安装前先为了避免后续权限不足导致操作失败和报错，可以先 sudo su 切换到root用户，然后 sudo apt update 、sudo apt upgrade 做下更新。

当然为了保证软件安装下载速度，请先切换到国内软件源 👉切换软件源

1、安装以下依赖

sudo apt-get install git cmake build-essential
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 libopenblas-dev liblapack-dev libatlas-base-dev  
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install git cmake build-essential

保证每条都安装成功，如果有依赖冲突再问度娘

2、下载caffe

使用命令 git clone https://github.com/BVLC/caffe.git 把caffe拉取到本地， 如果速度太慢的话，可以尝试直接进入页面下载，如下图：把链接https://github.com/BVLC/caffe.git复制到浏览器打开进行下载。

外部下载完成后记得解压到并且改文件夹名字为 caffe ，以下操作都是在 /home/你的用户名/ 目录下进行的，所以解压和下载的 caffe 最好就放这个目录下。

3、修改Makefile编译配置

先 cd caffe 进入caffe软件目录，输入 ls 我们可以看到有个 Makefile.config.example 文件，这个是官方给的编译配置，我们使用 cp Makefile.config.example Makefile.config 拷贝一份下来，得到一个 Makefile.config 编译配置，现在需要修改这个配置

这里为了方便使用图形界面的 gedit 没有图形界面的小伙伴可以使用 vim 。

sudo gedit Makefile.config

打开编辑器，如下图修改

然后再找到这个地方，用下面配置替换图中配置

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial

然后还有个地方需要注意下，避免后续报错找不到python包。

注意那个 /user/local/lib/......，原来是没有这个local的，然后我的机器上pip安装完依赖后，dist-package是在 /user/local/....下面，所以这里要改下，小伙伴可以输入 ls /usr/local/lib/python2.7/dist-packages 看下python的依赖是不是都默认装在这个local下，不在就不用改 。

改完后保存，然后开始安装python依赖

4、安装python依赖

先安装 pip

sudo apt-get install python-dev python-pip gfortran

安装完后记得更新下pip，避免报错

sudo pip install --upgrade pip

然后安装一些依赖库，这些信息在 caffe/python/requirements.txt 中，这里为了避免报错导致装不上，使用指定源的方式安装，这里指定豆瓣源

sudo pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn 要安装的模块名

要安装如下模块

Cython >=0.19.2
numpy >=1.7.1
scipy >=0.13.2
scikit-image >=0.9.3
matplotlib >=1.3.1
ipython >=3.0.0
h5py >=2.2.0
leveldb >=0.191
networkx >=1.8.1
nose >=1.3.0
pandas >=0.12.0
python-dateutil >=2.5.0
protobuf >=2.5.0
python-gflags >=2.0
pyyaml >=3.10
Pillow >=2.3.0
six >=1.1.0

然后用那条命令替换模块名称就可以了，当然你可以输入多个模块 空格隔开，这里我帮大家输入好了，直接复制

sudo pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn Cython numpy scipy scikit-image matplotlib ipython h5py leveldb networkx nose pandas python-dateutil protobuf python-gflags pyyaml Pillow six 

如果报 pandas 找不到版本信息则可以先忽略

5、配置环境变量

执行 sudo gedit ~/.bashrc 进行配置，在文件末尾另起一行加入下面内容，也就是指向/home/.../caffe/python/ 目录，注意替换你的用户名

export PYTHONPATH=/home/你的用户名/caffe/python:$PYTHONPATH

改完保存，执行 source ~/.bashrc 使配置生效，重启更稳妥

6、编译caffe

然后我们回到 caffe目录下

执行 sudo make all ，开始执行编译，编译有点慢，耐心等待。。。

编译如果成功则下面执行 sudo make pycaffe编译python caffe 接口

如果编译报错导致失败，则可以使用 sudo mke clean 清除后再次编译。

两个都编译成功的话，就可以直接进入python命令行交互，输入 import caffe 回车，如果没有任何输出则安装成功。

请注意： 如果有输出异常信息，请看下提示是不是说缺少什么模块，如果是缺少的话，用 sudo pip install 模块名 安装下，再尝试，如果报找不到 caffe 模块，就需要检查之前的步骤和配置后，执行make clean 然后再重新编译！！

有问题欢迎留言😀