Ubuntu 14.04 下 caffe 的安装
官方安装指导网站:
http://caffe.berkeleyvision.org/installation.html
1、General dependencies
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
2、CUDA安装
NVIDIA公司官网下载cuda最新版本runfile安装包,下载完以后使用md5sum校验下载文件是否正确,网址:
https://developer.nvidia.com/cuda-downloads
顺便把pdf的安装guide文件下载下来。
按照文档
Chapter 2. Pre-installation Actions
Chapter 6 Pre-installation actions
Chapter 4. Runfile Installation
顺序安装
3、BLAS: install ATLAS by
sudo apt-get install libatlas-base-dev
or install OpenBLAS or MKL for better CPU performance.
4、Python:系统默认自带Python2.7,但还需要安装python-dev
sudo apt-get install python-dev
5、其余的安装
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
以上安装完,下载caffe工程,就能够编译通过
cp Makefile.config.example Makefile.config
# Adjust Makefile.config (for example, if using Anaconda Python)
make all
make test
make runtest
可以采用-j8,加速编译如:
make all -j8
如果需要安装pycaffe,还需要安装python目录下,
requirements.txt
文件中列举的软件,root权限下使用以下命令安装(在python目录下执行)
for req in $(cat requirements.txt); do pip install $req; done
安装中会出错,提示缺少fortran编译器,需要先安装编译器
sudo apt-get install gfortran
安装完以后列表中软件就能全部安装成功。
6、安装pycaffe
在caffe工程根目录下,安装pycaffe
make pycaffe
可能会提示找不到numpy里面的头文件,这是因为安装numpy的目录和makefile.config文件里指定的目录不一样。默认目录为
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
但numpy可能在
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/local/lib/python2.7/dist-packages/numpy/core/include
修改makefile.config文件就能够成功安装pycaffe。
7*、draw_net.py依赖软件
python功能中,draw_net.py可以可视化网络结构,需要使用,要先安装pydot和graphviz两个软件
pip install pydot
apt-get install graphviz
over!