Caffe 学习系列(一):Caffe-CPU/GPU安装

Caffe 学习系列(一):安装

一.Caffe-CPU安装

参考这篇

可视化工具:Netscope Editor
网址:http://ethereon.github.io/netscope/#/editor
使用方法:将.prototxt中的内容输入到文本框中,然后按shift+enter键,就会得到可视化网络

1、准备工作
1. 1 安装依赖和相关科学计算库
sudo apt-get install libprotobuf-dev 

sudo apt-get install libleveldb-dev

sudo apt-get install libsnappy-dev 

sudo apt-get install libopencv-dev

sudo apt-get install libhdf5-serial-dev

sudo apt-get install protobuf-compiler

sudo apt-get install libgflags-dev

sudo apt-get install libgoogle-glog-dev

sudo apt-get install liblmdb-dev

sudo apt-get install libatlas-base-dev

sudo apt-get install python-pip

sudo apt-get install gfortran

sudo apt-get install python-protobuf

sudo apt-get install python-skimage

sudo apt-get install --no-install-recommends libboost-all-dev  

1. 2 安装编译工具 git cmake
sudo apt-get install git cmake
2、下载caffe

安装结束后下载caffe,执行以下命令:

# 克隆方式下载很慢,可直接进网站下载压缩包
git clone git://github.com/BVLC/caffe.git
cd caffe/
3、caffe配置与安装
3. 1 配置文件备份

养成良好习惯:在对源文件修改前,进行备份以免配置失误无法撤回
(1)对整个文件备份
(2)对文件内要修改的语句复制修改,注释掉默认的语句

生成Makefile.config文件,将caffe目录下自带的Makefile.config.example文件复制一份并更名为Makefile.config,命令如下:

cp Makefile.config.example Makefile.config
3. 2 配置文件修改

CPU_ONLY = 1               # 前面的 # 号去掉
WITH_PYTHON_LAYER := 1     # 前面的 # 号去掉

# OpenCV默认使用2.4.9,如果使用3.0版本做如下修改
# Uncomment if you're using OpenCV 3
OPENCV_VERSION := 3        # 前面的 # 号去掉()

配置引用文件路径(主要是HDF5的路径问题) ,加上如下文件如下

# whateber else you need goes here
INCLUDE_DIRS :=$(PYTHON_INCLUDE)/usr/local/include  /usr/include/hdf5/serial

LIBABRY_DIRS :=$(PYTHON_LIB) /usr/local/lib  usr/lib  /usr/lib/x86_64-linux-gnu/hdf5/serial

新增的内容为:

/usr/include/hdf5/serial
/usr/lib/x86_64-linux-gnu/hdf5/serial
3. 3 编译

make  -j8
make pycaffe
make test
make runtest

3.4 添加环境变量
cd ~/caffe/python
for req in $(cat requirements.txt); do pip install $req; done
sudo pip install -r requirements.txt

先打开配置文件bashrc

sudo gedit ~/.bashrc

在文件的最后面添加

export PYTHONPATH=/home/moqi/caffe/python:$PYTHONPATH

保存退出

sudo source ~/.bashrc
3.5 验证python接口
import caffe

未报错即安装成功

4、报错及解决办法

在python验证过程中出现以下错误:

numpy.dtype size changed, may indicate binary incompatibility

原因:各种库之间版本不匹配,将numpy版本降到1.14.5即可

sudo pip uninstall numpy
sudo pip install numpy==1.14.5

具体参考这篇

二.Caffe-GPU安装

1、安装过程

超详细配置Caffe(gpu版本+ubuntu16.04)考虑各种问题

Ubuntu16.04 Caffe 安装步骤记录(超详尽)
Ubuntu16.04+CUDA8.0+caffe配置
注意:安装 Cudnn 库时,注意 libcudnn.so.5.0.5 的后缀数字型号

 cd cuda;
 sudo cp lib64/lib* /usr/local/cuda/lib64/;
 sudo cp include/cudnn.h /usr/local/cuda/include/ 更新软连接:
 cd /usr/local/cuda/lib64/ 
 sudo chmod +r libcudnn.so.5.0.5   # 不同版本不同 本文是5.0.5
 sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5 
 sudo ln -sf libcudnn.so.5 libcudnn.so 
 sudo ldconfig
2、安装报错
  1. make runtest 时报错
    在 Makefile.config 中注释 USE_CUDNN 行。(暂时使用)

  2. ImportError: No module named google.protobuf.internal

sudo apt-get install python-protobuf 
  1. Import Error:cannot import name main
    pip升级后Import Error:cannot import name main解决方案

  2. caffe编译遇到的问题(持续更新)

3、拓展

Mobilenet-yolo

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

la_fe_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值