转载请注明出处:http://my.oschina.net/gaussik/blog/600368
参照github上happynear的caffe-windows项目,将Caffe for Windows配置成功,并且测试了其转换好的mnist数据库,不论是速度还是结果上,效果都相当好。现总结一下配置方法。
首先需要安装Visual Studio 2013。
然后再安装CUDA 7.5。注意先后顺序。
下载整个的caffe-windows项目,解压缩到所需目录下,例如本文中 D:\caffe。
下载作者制作的三方库并解压缩到项目的3rdparty目录。解压好后,将3rdparty/bin文件夹加入到环境变量PATH中,这样程序才能找到这些三方dll。、
GPU | Computer Capability |
---|---|
GTX660, 680, 760, 770 | compute_30,sm_30 |
GTX780, Titan Z, Titan Black, K20, K40 | compute_35,sm_35 |
GTX960, 970, 980, Titan X | compute_52,sm_52 |
你可以在https://en.wikipedia.org/wiki/CUDA上查看你的GPU所对应的Computer Capability。
点击工具栏绿色箭头进行编译,需要一定时间。
下载作者已经转换好的MNIST的leveldb数据文件,解压到./examples/mnist文件夹中,然后运行根目录下的run_mnist.bat即可开始训练,日志会保存在./log文件夹中。可看到,迭代10000次,准确率达到了0.9925。
若有更新,请到文中所给链接处实时查看。