VMware Workstation Pro + Ubuntu 16.04 + 上 Caffe 配置安装(Only CPU)

来源:互联网 发布:excel 一列数据递增 编辑:程序博客网 时间:2024/06/06 07:48

最近博主要准备着手用caffe进行人工智能方向的学习,殊不知败在了caffe环境搭建上,历经千辛万苦不断的Google终于把caffe搭建好了,下面就给出博主搭建caffe方法。

如果你是刚刚装好虚拟机,为了防止下载速度过慢,请用以下方法将Linux的下载源改为国内源。
1.登录系统

2.修改软件源:APT-sources.list

打开终端输入:

sudo gedit /etc/apt/sources.list

3.选择适合自己较快的源:

源链接:http://wiki.ubuntu.org.cn/index.php?title=%E6%BA%90%E5%88%97%E8%A1%A8&variant=zh-cn

4.对软件进行一次整体更新:

sudo apt-get update sudo apt-get upgrade

安装依赖包
1.安装protobuf,leveldb,snappy,opencv,hdf5, protobuf compiler andboost:

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compilersudo apt-get install --no-install-recommends libboost-all-dev

2.安装gflags,glogs ,lmdb andatlas.

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-devsudo apt-get install libatlas-base-dev

下载Caffe
使用git直接下载Caffe非常简单,或者去https://github.com/BVLC/caffe下载

 sudo apt-get install gitgit clone git://github.com/BVLC/caffe.git

编译Caffe
1.切换到Caffe所在目录

cp Makefile.config.example Makefile.config

配置Makefile.config

1)CPU_ONLY := 1

2)配置一些引用文件(增加部分主要是解决新版本下,HDF5的路径问题)

-两个路径要改成这样:(添加后面的两个hdf5的路径, 否则编译时报hdf5错误)
/# Whatever else you find you need goes here.
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/hdf5/serial
3)设置Python Caffe(将上面的路径改为下面的路径)
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include**

PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/local/lib/python2.7/dist-packages/numpy/core/include

编译python接口
1.Caffe拥有python\C++\shell接口,在Caffe使用python特别方便,在实例中都有接口的说明。

1)确保pip已经安装

sudo apt-get install python-pip

2)切换到Caffe/python所在目录,新建shell文件并执行安装依赖

for req in $(cat requirements.txt); do pip install $req; done

Make Caffe

sudo make pycaffe -j8sudo make all -j8sudo make test -j8sudo make runtest -j8

博主在Make Caffe这一步遇到了一万个错误,一般直接把错误信息复制到StackOverflow上都可以得到解答,一定要有耐心,相信自己一定可以编译通过的!
另外, 这个make默认是用CPU单核运算,如果想要快一点, 比如我想使用八核, 在make后面加上-j8标签.如果上面4行某一行报错之后想要重试,一定要先sudo make clean再重新开始

这时候应该可以了,试验一下:
切换到Caffe/python所在目录

$ pythonPython 2.7.12 (default, Jul  1 2016, 15:12:24) [GCC 5.4.0 20160609] on linux2Type "help", "copyright", "credits" or "license" for more information.>>> import caffe>>> 

说明安装全部完成!

参考:
[1]http://blog.csdn.net/u010402483/article/details/51506616 Caffe学习(1):Ubuntu16.04上Caffe配置安装(Only CPU)
[2]http://www.linuxidc.com/Linux/2016-09/135034.htm Ubuntu 16.04上安装Caffe(CPU only)
[3]http://blog.csdn.net/weixin_38034182/article/details/76672906 Linux系统软件安装更新下载太慢解决方法(更换国内源)

阅读全文
1 0
原创粉丝点击