$sudo docker run -ti elezar/caffe:cpu
获取Mnist数据集
$ cd /opt/caffe/data/mnist/
$ ./get_mnist.sh
对Mnist数据集进性训练
1、转换数据格式:
$sudo sh /opt/caffe/examples/mnist/create_mnist.sh
2、进行训练
$sudo sh /opt/caffe/examples/mnist/train_lenet.sh
若有错,需要更改lenet_solver.prototxt文件
用vi命令打开:$sudo sh /opt/caffe/examples/mnist/lenet_solver.prototxt
将末尾的GPU改为CPU
再次输入:
$sudo sh /opt/caffe/examples/mnist/train_lenet.sh
运行成功