您的位置:首页 > Web前端

Ubuntu 14.04 64bit caffe安装

2017-08-15 16:22 411 查看
本篇博客主要用于记录Ubuntu 14.04 64bit操作系统搭建caffe环境,目前针对的的是CPU版本;

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


CUDA:如果仅仅是cpu安装,忽略这一步;

BLAS:安装ATLAS
sudo apt-get install libatlas-base-dev


Python:可选

2.安装python依赖的依赖库:

安装gfortran,后面编译过程中会用到
sudo apt-get install gfortran


安装blas,Ubuntu下对应的是libopenblas,其它操作系统可能需要安装其它版本的blas——这是个OS相关的。
sudo apt-get install libopenblas-dev


安装lapack,Ubuntu下对应的是liblapack-dev,和OS相关。
sudo apt-get install liblapack-dev


安装atlas,Ubuntu下对应的是libatlas-base-dev,和OS相关。
sudo apt-get install libatlas-base-dev


安装pip
sudo apt-get install python-pip
sudo apt-get install python-dev
sudo apt-get install python-nose
sudo apt-get install g++
sudo apt-get install git


3.Ubuntu14.04安装相关依赖库
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev


4.通过git,拉取源码
git clone https://github.com/BVLC/caffe.git[/code] 
5.安装python依赖
cd caffe/python
for req in $(cat requirements.txt); do sudo pip install $req; done


6.编辑caffe所需的Makefile文件
cd caffe
cp Makefile.config.example Makefile.config
vim Makefile.config


主要修改这两处
#去掉注释
CPU_ONLY := 1

BLAS_INCLUDE := /usr/include/atlas/


7.编译caffe
make -j4


测试编译结果:
make test
make runtest


测试结果如下:

8.配置python
# 编译pycaffe
make pycaffe -j4
make distribute


进入python后,直接import caffe会失败,

在.bashrc中配置,
sudo gedit ~/.bashrc
# 我的caffe/python目录 /home/zhb/Desktop/Work/caffe/python
export PYTHONPATH=$PYTHONPATH:/your/path/caffe/python
source ~/.bashrc


9.mnist数据测试
cd caffe

#下载mnist数据
sh data/mnist/get_mnist.sh

sh examples/mnist/create_mnist.sh

# 修改 solver_mode 为 CPU
vim examples/mnist/lenet_solver.prototxt

./examples/mnist/train_lenet.sh


执行结果如下:
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: