Ubuntu16.04+caffe+digits安装配置

前端之家收集整理的这篇文章主要介绍了Ubuntu16.04+caffe+digits安装配置前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

注:本文主要介绍的是如何在ubuntu16.04系统下安装caffe以及可视化工具digits,至于cuda和cudnn的安装配置在我前一篇文章http://blog.csdn.net/cdwxx1234/article/details/75121562已经介绍了,此文不再重复。不多说了,我们开始吧!

一.必要依赖包安装

  1. sudoapt-getinstallbuild-essential
  2. sudoapt-getinstall--no-install-recommendslibboost-all-dev
  3. sudoapt-getinstalllibatlas-base-devlibgflags-devlibgoogle-glog-devliblmdb-dev
  4. sudoapt-getinstalllibprotobuf-devlibleveldb-devlibsnappy-devlibopencv-devlibhdf5-serial-devprotobuf-compiler
  5. sudoapt-getinstalllibblas-devliblapack-devlibatlas-base-devgfortranpython-numpy

二.安装pip和easy-install

  1. cd
  2. wget--no-check-certificatehttps://bootstrap.pypa.io/ez_setup.py
  3. sudopythonez_setup.py--insecure
  4. wgethttps://bootstrap.pypa.io/get-pip.py
  5. sudopythonget-pip.py

三.安装caffepython依赖

1).安装git

  1. sudoapt-getinstallgit

2).获取caffe源码(可cd到指定文件夹下下载):

  1. gitclonehttps://github.com/BVLC/caffe.git

3).在caffe根目录python文件夹下安装依赖项;

  1. sudoapt-getinstallpython-pip
  2. sudoforreqin$(cat"requirements.txt");dopipinstall-ihttps://pypi.tuna.tsinghua.edu.cn/simple$req;done

四.编译caffe

打开终端

  1. cdcaffe
  2. sudocpMakefile.config.exampleMakefile.config
  3. sudogeditMakefile.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

  1. makeall–j16(16线程,线程越多速度越快)
  2. maketest–j16
  3. makeruntest–j16
  4. makepytest–j16

一般在编译make all 的时候可能会出错,后面出错的几率不大。碰到的问题和解决方法与安装目标检测ssd遇到的一样,具体参考http://blog.csdn.net/cdwxx1234/article/details/75136657

编译成功后测试python

  1. cdcaffe/python
  2. python
  3. importcaffe

没有报错,编译就成功了。

五.安装digits

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

1).获取digits安装包

  1. sudogitclonehttps://github.com/NVIDIA/DIGITS.gitdigits

2).安装digits及依赖项

  1. cddigits
  2. sudoapt-getinstallgraphvizgunicorn
  3. forreqin$(catrequirements.txt);dosudopipinstall$req;done

3).打开digits

  1. cddigits
  2. ./digits-devserver

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

4).遇到的问题

在终端输入./digits-devserver后出现如下错误

ValueError: Caffe executable not found inPATH


解决方案:

1).检查当前envvar的值

  1. echo$CAFFE_ROOT
2).把envvar加到~/.profile中,下次当你登录时会自动加载
  1. echo"exportCAFFE_ROOT=/home/username(我的账户名)/caffe/">>~/.profile
3).加载新配置
  1. source~/.profile
4).检查新配置
  1. echo$CAFFE_ROOT
  2. /home/username/caffe/

按上面的方法改了之后,digits顺利启动。

猜你在找的Ubuntu相关文章