caffe+cuda9.1+cudnn7.0.5+digits6.1

版权声明:本文为博主原创文章,欢迎转载,但请注明出处 https://blog.csdn.net/lulugay/article/details/83316920

安装caffe+digits

先参照https://blog.csdn.net/lulugay/article/details/83316886 《Ubuntu16.04配置cuda+cuDNN》配置cuda+cudnn。

安装OpenBLAS以加速CPU推断速度

$ git clone https://github.com/xianyi/OpenBLAS.git
$ cd OpenBLAS
$ make OpenMP=1
$ sudo make install

默认安装位置是/opt/OpenBLAS

下载NVIDIA版caffe

$ git clone https://github.com/NVIDIA/caffe.git

安装caffe依赖的库

$ sudo pip install -r caffe/python/requirements.txt
$ cat caffe/python/requirements.txt | xargs -n1 sudo pip install
$ sudo apt-get install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf python-gevent python-h5py python-numpy python-pil python-pip python-scipy python-tk -y
$ sudo apt-get install libatlas-base-dev -y
$ sudo apt-get install --no-install-recommends build-essential cmake git gfortran libatlas-base-dev libboost-filesystem-dev libboost-python-dev libboost-system-dev libboost-thread-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libopencv-dev libsnappy-dev python-all-dev python-dev python-h5py python-matplotlib python-numpy python-opencv python-pil python-pip python-pydot python-scipy python-skimage python-sklearn -y
$ sudo apt-get install libboost-all-dev -y
$ sudo apt-get install libgoogle-glog-dev -y
$ sudo apt-get install libprotobuf-dev protobuf-compiler -y
$ sudo apt-get install python-tk -y
$ sudo apt-get install libturbojpeg -y
$ sudo ln -s /usr/lib/x86_64-linux-gnu/libturbojpeg.so.0.1.0 /usr/lib/x86_64-linux-gnu/libturbojpeg.so

配置caffe

将Makefile.config.example的内容复制到Makefile.config并修改

$ cp Makefile.config.example Makefile.config
$ vim Makefile.config

若使用cudnn,则将

#USE_CUDNN := 1

修改成:

USE_CUDNN := 1

若使用的opencv版本是3.*的,则将

#OPENCV_VERSION := 3 

修改成:

OPENCV_VERSION := 3

编译caffe

$ cd caffe
$ sudo mkdir build
$ cd build
$ sudo cmake ..
$ sudo make -j8
$ sudo make install

安装digits

下载digits

$ git clone https://github.com/NVIDIA/DIGITS.git
$ mv DIGITS digits
$ sudo pip install -r digits/requirements.txt
$ sudo pip install -e digits

测试digits

$ cd digits
$ ./digits-devserver

猜你喜欢

转载自blog.csdn.net/lulugay/article/details/83316920