dui

Win10下gpu1080ti caffe 跑cifar10例子详细步骤
1.cifar-10下载地址 http://www.cs.toronto.edu/~kriz/cifar.html
在这里插入图片描述
选择第三个下载!
我放在了H:\caffe-master下。
2.解压后把里面“.bin”后缀的文件拷贝到“H:\caffe-master\bin\input(自己创建一个\bin\input文件夹)”中。然后再创建一个“output”文件夹来存放下面转化的leveldb文件。
3.在“H:\caffe-master\bin”路径下创建一个名为cifar1.的bat文件,内容如下:
H:\caffe-master\Build\x64\Release\convert_cifar_data.exe input output leveldb
Pause
双击运行cifar1.bat,不出意外在“output”文件夹下会生成“cifar10_test_leveldb”和“cifar10_train_leveldb”两个文件夹。如下所示:
在这里插入图片描述

4.求数据图像的均值,获取“mean.binaryproto”均值文件.在“H:\caffe-master\bin”路径下创建一个名为cifar2的bat文件,内容如下:
H:\caffe-master\Build\x64\Release\compute_image_mean.exe --backend=leveldb.\output\cifar10_train_leveldbmean.binaryproto
Pause
双击运行bat,在“F:/caffe-master/bin”路径下会生成“mean.binaryproto”。如下图所示:
在这里插入图片描述
5.打开
H:\caffe-master\bin\examples\cifar10\cifar10_quick_train_test.prototxt”修改为部分一共为六处地方,如下:
第11行改为 mean_file: “H:/caffe-master/bin/mean.binaryproto”
第14行为 source: “H:/caffe-master/bin/output/cifar10_train_leveldb”
第16行改为 backend: LEVELDB
第28行改为 mean_file: “H:/caffe-master/bin/mean.binaryproto”
第31行改为 source: “H:/caffe-master/bin/output/cifar10_test_leveldb”
第33行改为 backend: LEVELDB

和H:\caffe-master\bin\examples\cifar10\“cifar10_quick_solver.prototxt”修改为部分一共为两处地方,如下:
第4行改为
net: “H:/caffe-master/bin/examples/cifar10/cifar10_quick_train_test.prototxt”
第24行改为
snapshot_prefix: “H:/caffe-master/bin/examples/cifar10/cifar10_quick”
6.在“H:\caffe-master\bin”路径下创建一个名为cifar3的bat文件,内容如下:
H:\caffe-master\Build\x64\Release\caffe.exe train --solver=.\examples\cifar10\cifar10_quick_solver.prototxt
Pause
然后双击运行。当显示Optimization Done.表示训练结束。GPU运行了1分钟左右,之前弱基用CPU跑了90分钟左右。
在这里插入图片描述
在这里插入图片描述

训练完成之后会在“H:\caffe-master\bin\examples\cifar10”下生成如下两个文件如下图所示:

至此完毕,收工睡觉!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值