caffe搭建
sudo chmod -R 777 /home/lc/caffe/
下面就是详细的安装过程(使用CPU,不使用GPU):
点击这里查看原文。
第一步:下载所需要的依赖库
在命令行输入以下命令。
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev sudo apt-get install libhdf5-serial-dev protobuf-compiler
sudo apt-get install libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
可能会有的问题解决:sudo apt-get update
第二步:下载caffe
git clone https://github.com/bvlc/caffe.git
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
出现问题:
提示git尚未安装。
解决办法:
sudo apt-get install git
重新git clone https://github.com/bvlc/caffe.git
(注意国内下载有些慢,需要等一些时间)
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
第三步:修改配置文件
cd caffe
mv Makefile.config.example Makefile.config
打开Makefile.config文件
将
# CPU_ONLY := 1
改为
CPU_ONLY := 1
第五步开始编译(加速编译)
make -j
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
如果提示无法找到 hdf5.h 请打开 Makefile.config 更改如下语句
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
如果提示无法找到 libhdf5.so 请打开 Makefile.config 更改如下语句
LIBRARY_DIRS := $(PYTHON_LIB) /usr/lib/x86_64-linux-gnu/hdf5/serial /usr/local/lib /usr/lib
之后编译 test 文件下的源码,同理输入以下命令即可。
make test
make runtest
第六步:测试
完成上述操作后即成功安装 Caffe,本实例使用 cifar10 来进行测试,在 caffe 目
录下依次执行以下操作:
输入以下命令:
./data/cifar10/get_cifar10.sh
./examples/cifar10/create_cifar10.sh
将 examples/cifar10/cifar10_quick_solver.prototxt 中的 GPU 改为 CPU;
将examples/cifar10/cifar10_quick_solver_lr1.prototxt中的GPU改为CPU;
输入命令:
./examples/cifar10/train_quick.sh
pycaffe 环境设置
Caffe 训练及测试时会用到 python 接口,下面介绍 pycaffe 接口的配置,其
流程如下:
下载库: 回到home
sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python3-pip python-pip
sudo pip install python-skimage
sudo apt-get update python-h5py python-protobuf python-leveldb python-networkx pythonnose python-pandas python-gflags
sudo pip install Cython ipython
cd caffe
make pycaffe
New a terminal
sudo gedit /etc/profile
最后一行添加export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH
source /etc/profile使其生效
Test
cd python
python
在Python中输入:import caffe
若不成功,可以
make clean
make pycaffe
是跟着游老师课程来的,第一次搭建的时候也没仔细看提供的文档,也不熟悉一些指令,导致废了很多功夫但毫无成效。后来虚拟机崩了,重新搭了一次,倒还是挺快的。
tips:注意记录一些操作和备份一下虚拟机