配置环境为WIN10 64位,在网上搜索了很多资料加以整理,并按自已实际配置过程完成的。
配置中主要参考了以下链接。
http://blog.csdn.net/guoyk1990/article/details/52909864
http://blog.csdn.net/zhj_matlab/article/details/69943869
http://blog.csdn.net/lynnandwei/article/details/43273077
以及其他一些未曾记下来的博客,在他们的帮助下才配置成功,此文章用作保留自已配置过程中的一些记忆。
第一步:下载所需软件
1、下载CUDA9.1,也是目前的最新版本。https://developer.nvidia.com/cuda-downloads
2、下载CUDNN5.1。https://developer.nvidia.com/rdp/cudnn-download
需要说明的是,尽管上面没有说明支持CUDA9.1,但是是可以使用的。起初我下载的是最新的v7.0.5,然而在配置过程中,会出现函数调用参数不匹配的问题,所以按照caffe中的要求下载CUDNN v5版本。
3、下载Caffe。
下载Caffe。https://github.com/BVLC/caffe/tree/windows
4、应已经安装好了VS2013,下载NuGet Package Manager。
https://marketplace.visualstudio.com/items?itemName=NuGetTeam.NuGetPackageManagerforVisualStudio2013
第二步:安装和配置
参考网上的大多数教程,对软件进行安装配置。
1、安装CUDA9.1,安装过程很简单。
2、CUDA9.1安装完成后,打开CUDNN文件,将里面的bin、lib、include三个文件夹复制到CUDA9.1的安装路径中与CUDA9.1本身存在的三个文件夹合并。当然,也可以直接在后面所述的工程配置中改变CUDNN的路径。
3、安装NuGet ,也可以在VS2013菜单中的 工具-扩展和更新-联机中搜索Nuget进行安装。
4、将caffe文件存放到一个文件夹,将caffe-master\windows\下的CommonSettings.props.example另存一份命名为CommonSettings.props,打开caffe-master\windows\Caffe.sln,然后打开工程中的配置项。
修改里面的相关信息,可参考所附链接。
修改CUDA版本为9.1,如下图所示。
启用NuGet 。
修改工程配置。
配置完成后重新生成解决方案,通过了则一步完成。
第三步 训练和测试手写体识别MNIST
1、下载MNIST数据,共四个文件。
http://yann.lecun.com/exdb/mnist/,将这四个文件放入caffe-master\examples\mnist\mnist_data中,参照链接1,可以依照其步骤进行。
在上行路径下,建立两个bat文件,convert-mnist-data-test.bat和convert-mnist-data-train.bat。分别保存以下命令。
convert-mnist-data-test.bat中
Build\x64\Release\convert_mnist_data.exe --backend=lmdb examples\mnist\mnist_data\train-images.idx3-ubyte examples\mnist\mnist_data\train-labels.idx1-ubyte examples\mnist\mnist_data\mnist_train_lmdb
pause
convert-mnist-data-train.bat中
Build\x64\Release\convert_mnist_data.exe --backend=lmdb examples\mnist\mnist_data\t10k-images.idx3-ubyte examples\mnist\mnist_data\t10k-labels.idx1-ubyte examples\mnist\mnist_data\mnist_test_lmdb
Pause
分别双击运行,应该生成了两个文件夹,将此两个文件夹复制到caffe-master\examples\mnist中。
2、在caffe-master中新建两个bat文件run和test。
run中:
Build\x64\Release\caffe.exe train --solver=examples/mnist/lenet_solver.prototxt
pause
test中:
Build\x64\Release\caffe.exe test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_10000.caffemodel -gpu=0
pause
运行run.bat进行训练的结果为,
运行test.bat测试的结果为