【Caffe学习】windows下caffe（GPU）的安装与配置

前言

安装环境

操作系统：Windows 10 企业版 LTSC

IDE：Microsoft Visual Studio 2013 Ultimate

CUDA：CUDA8.0

CUDNN：CUDNN6.0.21

一、准备工作

下载cffe
Windows平台下有两个caffe版本，一个是caffe提供的Windows工具包，另一个是微软提供Windows工具包，本文使用的是微软提供的Windows工具包。
Github： https://github.com/Microsoft/caffe
百度云：链接：https://pan.baidu.com/s/1Ki5XVp44XmEmb3kK8z0YnA 提取码：cuda

二、配置项目

1. 修改配置文件
   1）将caffe-master\windows下的CommonSettings.props.example复制一份并重命名为CommonSettings.props
   2）<CudaVersion> 中的7.5替换为8.0、<CuDnnPath>中填写cudnn-8.0-windows10-x64-v6.0的地址
   3)加载libcaffe和test_all项目，如果加载不成功请，检查CommonSettings.props文件中<CudaVersion>是否改成8.0
2. 配置项目
   1）将libcaffe设置为启动项目
   2）将警告视为错误修改为否，右键–>属性–>配置属性–>C/C++ -->将警告视为错误改成否
   
   3）将cudnn.h文件第114行添加参数 dataType<Dtype>::type
   

三、项目生成

1)生成libcaffe
时间可能会有点漫长，如果生成失败，请检查之前步骤是否完全正确
2)生成caffe

四、使用mnist测试

1）下载minset数据集
本文使用的是直接生成的mdb文件
链接：https://pan.baidu.com/s/14QhvnYVW4Rz5H89yyVt7Ag 提取码：cuda
将下载好的文件解压到examples/mnist/下或将lenet_train_test.prototxt文件中的source改为自己的路径
2）训练模型
在项目路径下进入控制窗口，运行指令。

.\Build\x64\Debug\caffe.exe train --solver=examples/mnist/lenet_solver.prototxt  

训练完成后得到模型，位于caffe-master\examples\mnist\下，caffemodel文件
3）测试模型

.\Build\x64\Debug/caffe.exe test -model examples/mnist/lenet_train_test.prototxt -weights examples/mnist/lenet_iter_10000.caffemodel