您的位置:首页 > 编程语言 > Python开发

caffe 无cuda安装过程及python接口

2017-03-15 16:38 531 查看
主要过程稍微记录一下:

1.安装BLAS

sudo apt-get install libatlas-base-dev
2.安装依赖项

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev protobuf-compiler liblmdb-dev
3.安装gflags,glog和lmdb

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler
4.下载Caffe

git clone git://github.com/BVLC/caffe.git
5.安装Caffe

cd caffe
cp Makefile.config.example Makefile.config
因为这里没有gpu,所以需要设置Makefile.config文件中的CPU_ONLY:= 1,把这句的注释去掉就可以了。

然后编译

make all
make test
make runtest
可能会报错 我的缺少protobuf  下载地址 https://github.com/google/protobuf
注:



解决办法

将 Makefile.config文件中

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改为

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

cd /usr/lib/x86_64-linux-gnu
sudo ln -s libhdf5_serial.so.10.1.0 libhdf5.so
sudo ln -s libhdf5_serial_hl.so.10.0.2 libhdf5_hl.so



安装好以后我们就可以试着在mnist上跑一下lenet了。

1.首先获取mnist数据

cd caffe
./data/mnist/get_mnist.sh
2.然后创建lenet

./examples/mnist/create_mnist.sh
注意一定要在caffe的根目录下运行以下命令,否则会报“ build/examples/mnist/convert_mnist_data.bin: not found”的错误

3.训练cnn

没有gpu的话要记得把caffe/examples/mnist/lenet_solver.prototxt中的solver_mode设置成solver_mode: CPU。然后在根目录下执行

./examples/mnist/train_lenet.sh




准确率可以达到0.9912

因为caffe的tutorial上有很大一部分是Python的,所以后来又安装了一下Python的接口。

1.首先安装python

ubantu自带 有

2.安装pip

sudo apt-get install python-pip python-dev build-essential
3.运行以下代码安装必要的依赖项:

sudo pip install -r ./python/requirements.txt
4.在caffe的根目录下运行

make clean
make all
make pycaffe
这里遇到了一个问题:

virtual memory exhausted: Cannot allocate memory
make: *** [python/caffe/_caffe.so] Error 1
按照这里的方法增加Linux虚拟机的内存就可以解决了。

另一个问题



解决办法

sudo apt-get install python-numpy
sudo make pycaffe -j16
make pycaffe
5.把caffe/python的路径加到python路径中

运行python进入python shell,然后运行下列命令

import sys
sys.path.append("path/to/caffe/python/")  注:这是caffe中python路径
exit()
6.再次进入python shell,运行import caffe

注:不行的话重启一下电脑

文章主要参考:http://www.cnblogs.com/sunshineatnoon/p/4535329.html
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐