官方版caffe-Windows的安装与测试

文章转自:http://blog.sina.com.cn/s/blog_15e1762760102w9kc.html

笔者前几天无意中发现,深度学习框架caffe-windows出了官方版本 ,对于众多习惯使用windows机器的caffe使用者来说这无疑是福音,而且据说官方版的caffe-windows是由微软公司负责维护的哦~

废话不多说,下面就开始下载并配置吧。

第一步,准备工作

1.首先从网址下载Caffe-Windows安装包

2.安装Visual Studio 201;

3.有NVIDIA显卡的童鞋请安装​CudaCuDNN

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配置成功了。

猜你喜欢

转载自blog.csdn.net/qq_36621927/article/details/80788143