Caffe 安装

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
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install libatlas-base-dev

2. 获取源码

git clone https://github.com/BVLC/caffe.git

3.设置Caffe Python接口

cd caffe/python
for req in $(cat requirements.txt); do sudo pip install $req; done

还需要添加环境变量

sudo gedit ~/.bashrc

在文件的最后添加

#caffe python
export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

环境变量进行更新

cd 
source ~/.bashrc

4.修改配置文件

修改Makefile.config 文件

第一处
USE_CUDNN := 0

改为

#启用cuDNN加速
USE_CUDNN := 0
第二处
CUDA_DIR := /usr/local/cuda

改为

CUDA_DIR := /usr/local/cuda-8.0
第三处
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include 
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 

改为

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/

5.安装

make all -j
make test -j
make runtest -j

上边的 -j表示使用所有的CPU线程来进行安装,也可以在j后边添加自己想使用线程的数量。
###5.添加Caffe环境变量
还需要添加环境变量,在终端中输入环境变量

sudo gedit ~/.bashrc

在文件的最后添加

#Caffe
export CAFFE_ROOT=/home/liu/sfw/dlapp/caffe/

环境变量进行更新

cd 
source ~/.bashrc

6.安装成功检验

在终端中进入python运行环境

python

当终端进入python环境后,

import caffe

当程序没有报错时,caffe 1 就安装成功了。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值