您的位置:首页 > Web前端

安装 Caffe , Berkely的深度学习框架

2015-05-25 18:30 399 查看

安装 Caffe , Berkely的深度学习框架

标签(空格分隔): vision deepLearning

Caffe是Berkeley的一个深度学习框架,方便构造深度学习网络,并在GPU上运行。

安装过程按照Caffe的主页上的描述:http://caffe.berkeleyvision.org/install_apt.html

基本系统信息 Basic system information

使用Ubuntu Linux 14.04桌面版:

~$ uname -a
Linux cis-ubuntu-1 3.13.0-24-generic #46-Ubuntu SMP Thu Apr 10 19:11:08 UTC 2014 x86_64 x86_64 x86_64 GNU/Linux


需要安装的基础软件

caffe依赖BLAS等计算包, Boost 1.55以上,Opencv2.4以上。还有些包不太清楚是干什么的:glog(google认证用的?),hdf5(还可以用在hadoop的HDFS上?)。

直接用apt-get来安装:

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler


how to check?

检查Boost是否安装正确:

[[]https://www.zybuluo.com/huangzhengsjtu/note/105805][我写的如何安装Boost]

安装Cuda

下载Cuda 7.0版本for Linux的deb文件,然后安装。

wget http://developer.download.nvidia.com/compute/cuda/7_0/Prod/local_installers/rpmdeb/cuda-repo-ubuntu1404-7-0-local_7.0-28_amd64.deb sudo dpkg -i cuda-repo-ubuntu1404-7-0-local_7.0-28_amd64.deb


安装Caffe

从Githut下载caffe-master.zip. https://github.com/BVLC/caffe

解压缩,并且创建Makefile.config文件。

unzip caffe-master.zip
cd caffe-master
cp Makefile.config.example Makefile.config


编辑Makefile.config

例如:For cuDNN acceleration, you should uncomment the USE_CUDNN := 1 switch in Makefile.config.

例如:For CPU-only Caffe, uncomment CPU_ONLY := 1 in Makefile.config. (现在没有GPU,所以我就使用CPU)

例如:以后要运行Fast R-CNN,所以 WITH_PYTHON_LAYER := 1

Build

make all
make test
make runtest
[----------] Global test environment tear-down
[==========] 651 tests from 115 test cases ran. (36247 ms total)
[  PASSED  ] 651 tests.


实际例子

下一步在一个具体的例子上运行caffe,就选流行的minist数据集吧。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: