关键词

win10+vs2013+cuda8.0+caffe

配置:

环境:windows10,vs2013 community,cuda8.0,caffe,cudnn4
注意:先要安装好显卡驱动(我的显卡是1070),这里的例子只开通了matlab接口,需要用python可以安装anacoda并打开python接口。

2,准备文件

例子中用的文件:链接:http://pan.baidu.com/s/1hsBNI3i 密码:cynd
注意一:处理cudnn有两种方式,其一是在vs的属性文件中给出目录;另一是将cudnn的文件放到C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0的相应目录中;这里我用的是第二种方式。
注意二:解压出来的caffe-master应该有两层目录名字都是caffe-master,第一次编译的时候我直接去掉了一层,(因为我的小小强迫症),但是发现其实程序会在第一层caffe-master的下面新建一个NugetPackages文件夹来放置第三方库,所以实际上这里不要去掉其中一层caffe-master目录。
注意三:先编译libcaffe,再编译caffe,最后需要什么工程就编译哪个,编译过程中的错误会之后找个时间写一篇专门的博客。

3,测试mnist

需要下载的四个训练和测试数据已经在前面的分享文件夹中,下载即可;
转化数据格式到lmdb,用的是python代码,注意代码应该放在Build同目录下:

上图就能看到测试程序的结果,正确率等参数。祝成功,有问题留言!

本文链接:http://task.lmcjl.com/news/12841.html

展开阅读全文