Ubuntu16.04 + Digits + caffee

Posted 2020-12-28 pprp

reference: csdn

必要依赖包安装

sudo apt-get install build-essential
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install libblas-dev liblapack-dev libatlas-base-dev gfortran python-numpy

安装pip和easy-install

cd
wget --no-check-certificate https://bootstrap.pypa.io/ez_setup.py
sudo python ez_setup.py --insecure
wget https://bootstrap.pypa.io/get-pip.py
sudo python get-pip.py

安装caffe及python依赖

sudo apt-get install git
git clone https://github.com/BVLC/caffe.git
sudo apt-get install python-pip
sudo for req in $(cat "requirements.txt"); do pip install -i https://pypi.tuna.tsinghua.edu.cn/simple $req; done

编译caffe

cd caffe
sudo cp Makefile.config.example Makefile.config
sudo gedit Makefile.config

在打开的Makefile.config文件中做如下修改；

1).将USE_CUDNN := 1取消注释；

2).INCLUDE_DIRS := $(PYTHON_INCLUDE)/usr/local/include空格后然后添加/usr/include/hdf5/serial /usr/local/cuda-8.0/include/(添加cuda的include路径)；

3).LIBRARY_DIRS:=$(PYTHON_LIB)/usr/local/lib /usr/lib 空格后添加 /usr/local/cuda-8.0/lib64/(添加cuda的lib路径)；

开始编译caffe

make all –j16   （16线程，线程越多速度越快）
make test –j16
make runtest –j16
make pytest –j16

编译成功后测试python

cd caffe/python
python 
import caffe 

安装digits

digits是caffe的一个可视化工具，可使我们对caffe的操作变得方便容易。

sudo git clone https://github.com/NVIDIA/DIGITS.git digits


cd digits
sudo apt-get install graphviz gunicorn
for req in $(cat requirements.txt); do sudopip install $req; done 

cd digits
./digits-devserver

在网页上输入http://localhost:5000即可打开界面。