VS2013 + CUDA9.1 + CUDNN5.1 + Caffe配置过程

 配置环境为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测试的结果为










 

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值