ubuntu caffe 安装设置

来源:互联网 发布:js数组内数组拆分 编辑:程序博客网 时间:2024/06/05 09:15

安装插件

sudo apt-get install git

sudo apt-get install vim

sudo apt-get install build-essential

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

sudo apt-get install libatlas-base-dev

sudo apt-get install python-dev

下载caffe
git clone https://github.com/bvlc/caffe.git

保存更改make文件

cd /caffe-marster/

sudo cp Makefile.config.example Makefile.config

打开Makefile

vim Makefile.config       

讲CPU设为开启,去掉前面的#号

 CPU_ONLY = 1


编译
make -j2

测试
make test

make runtest


MMIST训练

cd ~/caffe


下载MNIST数据库并解压缩

./data/mnist/get_mnist.sh


将其转换成Lmdb数据库格式

./examples/mnist/create_mnist.sh


将文件/examples/mnist/lenet_solver.prototxt的最后一项改成CPU


训练网络
./examples/mnist/train_lenet.sh
0 0
原创粉丝点击