用docker快速部署caffe

1. 快速搭建 Day 4 环境:


从官方拉取 docker image$ docker pull zhaoyongke/bookenv:caffe_cpu_only如果速度很慢,推荐阿里云 Docker Hub:$ docker pull registry.cn-hangzhou.aliyuncs.com/master_caffe_in_21days/caffe:cpu_only


创建一个 docker 实例:

$ docker run -ti registry.cn-hangzhou.aliyuncs.com/master_caffe_in_21days/caffe:cpu_only

创建成功后,依次执行如下命令来运行 MNIST 例程。

切到 Caffe 目录。该源码克隆于 master 分支(20161112)。

$ cd root/caffe/$ ./build/tools/caffe.bin$ vi examples/mnist/lenet_solver.prototxt
将 solver_mode: GPU 改为 CPU
安装 wget,便于执行获取数据脚本$ apt-get install wget
获取 MNIST 数据集$ ./data/mnist/get_mnist.sh
用 MNIST 数据集创建 LMDB 数据库$ ./examples/mnist/create_mnist.sh

执行训练$ ./examples/mnist/train_lenet.sh


3.3.1 docker commit
$ docker commit c5f24e953610 mycaffe
 
 
  • 1
  • 1

其中c5f24e953610是我们之前所使用容器的ID,可以通过docker ps -a查看;mycaffe是新生成的镜像的名称。 

下载好了之后要把数据转换成caffe可以运行的LMDB格式。

./examples/mnist/create_mnist.sh
 
 
  • 1

然后bug就来啦:

$ sudo sh examples/mnist/create_mnist.sh
Creating lmdb...
./create_mnist.sh: 16: ./create_mnist.sh: build/examples/mnist/convert_mnist_data.bin: not found
./create_mnist.sh: 18: ./create_mnist.sh: build/examples/mnist/convert_mnist_data.bin: not found
Done.
 
 
  • 1
  • 2
  • 3
  • 4
  • 5

首先google到这个issue, 
When ./create_mnist.sh, fault: convert_mnist_data.bin: not found #1251 
主要说的还是要在caffe的根目录下运行。然而这对我并没有用。我进入build/examples/mnist/下面发现是真的没有convert_mnist_data.bin。这个时候我以为是编译的时候没有弄好,又去编译啦一遍,还是不行。于是又开始了漫长的google之路。最后找到啦这篇博客 
深度学习框架Caffe在Mac上的安装和测试 
它里面也是这个问题。它解决的方法是在文件中搜索convert_mnist_data.bin。然后修改路径。

于是我也在caffe的目录下搜索convert_mnist_data.bin,发现在./examples/mnist下面,于是将examples/mnist/create_mnist.sh中的build路径从

BUILD=build/examples/mnist
 
 
  • 1

改成

BUILD=examples/mnist
 
 
  • 1

或者直接把convert_mnist_data.bin移到build/examples/mnist下面。

然后就运行啦

 ./examples/mnist/train_lenet.sh


参考:http://www.th7.cn/system/lin/201611/189562.shtml

http://blog.csdn.net/elaine_bao/article/details/53117676

http://blog.csdn.net/wangxing233/article/details/51734080

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值