install caffe

Posted on |

这本课程中本人自学了caffe框架的用法。caffe是个很好的工具，特别是针对图像处理相关的卷积神经网络。但是本人的电脑系统为macOS，在参照官方文档时候出现问题并得不到解决，似乎是因为系统版本的缘故。后来就索性安装了一个ubuntu的虚拟机，分配了4G内存，两个core。跑官方mnist的例子需要13分钟左右。
caffe在ubuntu系统下的安装比较简单，因为显卡问题我没有装GPU，安装基本参照下面的链接网址。安装时候请注意最好换个源下载，因为官方的源速度实在太慢了。装了大半天基本都是在等待下载。换源方法

Install with CPU

安装依赖库

1
2
3
4
5
6
sudo apt update
sudo apt upgrade
sudo apt install -y build-essential cmake git pkg-config
sudo apt install -y libprotobuf-dev libleveldb-dev libsnappy-dev \
                    libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt install -y --no-install-recommends libboost-all-dev

如果使用 OpenBlas 代替默认的 ATLAS的话，需要将 libatlas-base-dev 改为 libopenblas-dev

1
2
3
sudo apt install -y libatlas-base-dev 
sudo apt install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt install -y python-pip python-dev python-numpy python-scipy

修改配置文件

1
2
3
4
5
cd ~/
git clone https://github.com/BVLC/caffe.git
cd caffe/
cp Makefile.config.example Makefile.config
vim Makefile.config
  • 取消注释 
1
2
CPU_ONLY := 1    
WITH\_PYTHON\_LAYER := 1
  • 如果选择 CPU 安装的话，最好使用 OpenBlas 代替默认的 ATLAS，因为 OpenBlas 对 CPU 多线程支持很好，能加快一点速度算一点吧。修改
    BLAS := open
  • 并运行以下命令（使用 OpenBlas 的情况下）
    echo 'export OPENBLAS_NUM_THREADS=4' >> ~/.bashrc
  • 修改

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 /usr/lib/x86_64-linux-gnu/hdf5/serial

安装 make pycaffe 所需的库

1
2
cd python
sudo -H pip2 install -r requirements.txt

安装

1
2
3
4
5
6
7
8
9
cd ..
make all -j $(($(nproc) + 1))
make test
make runtest
make pycaffe
make distribute
#user 是你的用户名
echo 'export PYTHONPATH=/home/user/caffe/python:$PYTHONPATH' >> ~/.bashrc

重启终端

1
2
3
python
>>>import caffe 
>>>               # 没报错就成功了

至此，caffe就算安装完成了，接下来就愉快的开始使用caffe吧。

参考网址

http://hmybmny.com/2016/10/install-caffe-on-ubuntu-16-dot-04-with-gpu/