文章转自:http://blog.sina.com.cn/s/blog_15e1762760102w9kc.html
笔者前几天无意中发现,深度学习框架caffe-windows出了官方版本 ,对于众多习惯使用windows机器的caffe使用者来说这无疑是福音,而且据说官方版的caffe-windows是由微软公司负责维护的哦~
废话不多说,下面就开始下载并配置吧。
第一步,准备工作
1.首先从网址下载Caffe-Windows安装包;
2.安装Visual Studio 201;
4.将.\windows\CommonSettings.props.example
复制一份并修改为 .\windows\CommonSettings.props,然后打开做如下修改:
1)如果不使用GPU,不安装CUDA和CuDNN,将
CpuOnlyBuild
改为 true
,并且 UseCuDNN
改为 false;
2)安装CuDNN的情况下,将CuDNND路径复制到
CuDnnPath;
3)选择支持Python或者MATLAB接口,当然也可以两个都支持,设置PythonSupport或者MatlabSupport为true,同时将Python或者MATLAB路径分别粘过来PythonDir和MatlabDir。
第二步,编译
打开.\windows\Caffe.sln,编译,至少要将Caffe/convert_imageset编译,其他可以根据需要进行,生成的caffe.exe为训练与测试的主程序,convert_imageset为将训练和测试数据生成LMDB使用的程序。
如果使用MATLAB的接口,将matcaffe编译,然后\Build\x64\Release设置到你的环境变量,同时将文件夹matcaffe添加到MATLAB搜索路径。
如果使用 Python的接口,将pycaffe编译,然后将\Build\x64\Release\pycaffe\caffe复制到\lib\site-packages即可以使用。
第三步,测试
下载MNIST数据库,解压缩后将mnist-test-leveldb与mnist-train-leveldb文件夹放到\examples\mnist下(如果不使用GPU运算,请打开\examples\mnist\lenet_solver.prototxt将solver_mode改为CPU),然后回到caffe根目录下,编写批处理文件run.bat内容如下:
Build\x64\Release\caffe.exe train --solver=examples/mnist/lenet_solver.prototxt
pause
双击run.bat程序能够运行,并且之后也不会报错,那么恭喜你,Caffe-Windows配置成功了。