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”下生成如下两个文件如下图所示:
至此完毕,收工睡觉!