@huangzhengsjtu
2015-05-22T08:49:14.000000Z
字数 1357
阅读 4236
vision
deepLearning
Caffe是Berkeley的一个深度学习框架,方便构造深度学习网络,并在GPU上运行。
安装过程按照Caffe的主页上的描述:http://caffe.berkeleyvision.org/install_apt.html
使用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
下载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
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)
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数据集吧。