忠于原著的原汁原味Caffe安装 Ubuntu+Windows双版本

来源:互联网 发布:linux 打包tar命令 编辑:程序博客网 时间:2024/04/30 14:40

前言

       本人小硕一枚,马上就要开始找工作,看着身边的人都有自己的blog,也随大流一下,顺便也总结下自己一年来的工作,个人水平一般,权当自娱自乐,如果能给你带来些许帮助当然更好。第一篇就写安装Caffe,虽然网上都写烂了,我还是写一篇自己安装的过程,万一哪个啥呢。。。

准备工作

  1. Caffe官网 http://caffe.berkeleyvision.org
  2. Caffe-Micrsoft https://github.com/BVLC/caffe/tree/windows
  3. CUDA https://developer.nvidia.com/cuda-downloads
  4. cuDNN https://developer.nvidia.com/cudnn
  5. Adaconda https://www.continuum.io/downloads
       首先本人不推荐没有N卡的同学整Caffe,因为最多就能训一个MNIST网络,实用性不大,如果是有钱有时间顶级CPU一大把的同学,算我什么都没说。

一、Ubuntu

本文完全按照Caffe官网提供的Ubuntu Installation上提供的步骤,添加了部分官网没有给出的操作 。

   1.安装依赖

  1. 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
安装Caffe框架所需要的依赖项,注意已经提供了Opencv如果没有其他的特别需要不用另外安装,网上很多帖子还推荐另外安装Opencv3.0,个人不是很推荐,官方提供的基本满足需要。

2.CUDA安装


本人刚开始的时候下载的是CUDA的.run文件,安装过程也是各种折腾,后来看了欧新宇老师的帖子,上面推荐使用.deb的离线包,各种方便,免去修改blacklist-nouveau等各种繁琐的操作
你需要在CUDA的官网下载最新的CUDA deb文件如下图,网址见上面的准备工作,而且图片的最下方也提供了安装步骤,这里怕图裂提供文字版:
sudo dpkg -icuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb`(下载的文件名)
sudo apt-get update`
sudo apt-get install cuda`

3.BLAS安装

Caffe支持多种Blas库,原生是ATLAS,笔者因为懒得去申请MKL,也就用的原生官网一句代码:
sudo apt-get install libatlas-base-dev

4.Python安装

推荐安装Adaconda这个集成库,基本上已集成了大部分的Python第三方包,不需要在额外添加,网址也在最上面的准备工作。同样一句代码:
bash <downloaded file>

5.环境注册

很多同学无法正常安装Caffe就是因为环境变量没有注册导致无法链接到或者找到文件,虽然有很多种注册的方法,但笔者还是喜欢简单粗暴的直接修改文件。
(1)修改environment文件,添加cuda和adaconda的bin文件路径(根据你的安装路径调整),注意有冒号,
sudo gedit /etc/environment:/usr/local/cuda-7.5/bin:/home/your name/anaconda2/bin:source /etc/environment

(2)修改.bashrc文件
sudo gedit ~/.bashrcexport PATH=/usr/local/cuda-7.5/bin:$PATHexport PATH=/home/your name/anaconda2/bin:$PATHsource ~/.bashrc

(3)修改ld.so.conf 文件
sudo gedit /etc/ld.so.conf  /usr/local/cuda-7.5/lib64/home/your name/anaconda2/libsudo ldconfig

6.glog gflags lmdb 安装

还是一条命令简单粗暴:
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

7.Caffe安装

从Github下载Caffe的Windows分支,它其实是双系统版本,Ubuntu和Windows都可以使用,解压,这时如果我们开始编译工作一般会出现权限问题,但全程使用root权限也不安全,我们只需要将caffe这个文件夹的权限提高,假设caffe解压在home文件夹下名称为caffe,我们只需要在home下执行:
sudo chmod 777 -R caffe
提升权限后我们执行下一步操作:
     cp Makefile.config.example Makefile.config
由于我们使用和Adaconda我们需要修改几行的内容,红色为修改前,紫色为修改后
# NOTE: this is required only if you will compile the python interface.
# We need to be able to find Python.h and numpy/arrayobject.h.
# PYTHON_INCLUDE := /usr/include/python2.7 \
#/usr/lib/python2.7/dist-packages/numpy/core/include
# Anaconda Python distribution is quite popular. Include path:
# Verify anaconda location, sometimes it's in root.
ANACONDA_HOME := $(HOME)/anaconda2
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
$(ANACONDA_HOME)/include/python2.7 \
$(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \

# We need to be able to find libpythonX.X.so or .dylib.
# PYTHON_LIB := /usr/lib
# PYTHON_LIB := $(ANACONDA_HOME)/lib
       
      最后:
 make all -j8 make test -j8 make runtest -j8

      需要python接口的:

 make pycaffe

二、Windows


因为是Micrsoft出品,windows下安装实在是太傻瓜,我就不写了,再贴下网址,安装请看README.md.

    
第一次写Blog,废话太多,写得很烂,接下来会把自己用caffe做过的一些项目和阅读caffe代码过程中的一些理解贴上了。

0 0
原创粉丝点击