#caffe版本:BVLC-caffe-windows
# 环境:windows 10\CUDA8.0\cuDNN-v5.1\VS2013
1、GPU计算环境
(1)下载并安装CUDA8.0;
(2)下载cuDNN。载后解压会有三个文件夹bin、include、lib。将这三个文件夹复制到cuda的安装目录中:\NVIDIA GPU ComputingToolkit\CUDA\v8.5。(cuda的安装目录中也有这三个文件夹,将这三个文件夹分别与原来存在的文件夹合并)。
2、caffe-windows
(1)下载官方caffe-windows并解压,将 .\windows\CommonSettings.props.example改名为CommonSettings.props;
(2)编译caffe-windows
用vs2013打开.\windows\Caffe.sln 并将解决方案的配置改为release,点菜单栏上的“生成->生成解决方案”,会将整个项目全部生成,这个时间会比较长(由于官方caffe-windows 的版本使用了NuGet管理第三方开发包,所以需要在vs2013上安装NuGet)。生成成功之后的文件都在.\Build\x64\Release中。
【PS】编译过程中可能遇到的错误:
a)errorC2220: 警告被视为错误 - 没有生成“object”文件 (..\..\src\caffe\util\math_functions.cpp)。
解决方法:
打开VS->项目->属性->配置属性->C/C++->常规->将警告视为错误修改为“否”,重新编译就好。
b)错误 5 error MSB4062: 未能从程序集 S:\NugetPackages\OpenCV.2.4.10\build\native\\private\coapp.NuGetNativeMSBuildTasks.dll 加载任务“NuGetPackageOverlay”。未能加载文件或程序集“file:///S:\NugetPackages\OpenCV.2.4.10\build\native\private\coapp.NuGetNativeMSBuildTasks.dll”或它的某一个依赖项。系统找不到指定的文件。 请确认 <UsingTask> 声明正确,该程序集及其所有依赖项都可用,并且该任务包含实现 Microsoft.Build.Framework.ITask 的公共类。
解决方法:
进入libcaffe的属性——》生成事件——》预先生成事件——》在生成中使用——》否,生成事件——》后期生成事件——》在生成中使用——》否
删除了NugetPackages文件夹下的OpenCV2.4.10文件夹了,然后进入项目的NuGet包还原重新还原OpenCV2.4.10包,再重新生成解决方案。
3、MNIST测试
(1)下载MNIST数据集,MNIST数据集包含四个文件,如下表所示:
train-images-idx3-ubyte.gz 训练集图片 - 55000 张 训练图片, 5000 张 验证图片
train-labels-idx1-ubyte.gz 训练集图片对应的数字标签
t10k-images-idx3-ubyte.gz 测试集图片 - 10000 张 图片
t10k-labels-idx1-ubyte.gz 测试集图片对应的数字标签
下载完后解压得到对应的四个文件,这四个文件不能直接用于caffe的训练和测试。需要利用生成的convert_mnist_data.exe把四个文件转换为caffe所支持的leveldb或lmdb文件。
(2)转换 训练\测试数据
a) 将四个文件放到 caffe-windows\data\mnist 文件夹下。然后运行脚本:get_mnist.sh;
b) 在caffe-windows\examples\mnist 目录下,新建一个convert-mnist-data.bat文件转换为训练数据,并在文件中添加代码:
set DATA=../../data/mnist
set TOOLS=../../Build/x64/Release
REM set BACKEND=lmdb
set BACKEND=leveldb
echo "Creating %BACKEND%..."
rd /s /q "mnist_train_%BACKEND%"
rd /s /q "mnist_test_%BACKEND%"
"%TOOLS%/convert_mnist_data.exe" %DATA%/train-images-idx3-ubyte %DATA%/train-labels-idx1-ubyte mnist_train_%BACKEND% --backend=%BACKEND%
"%TOOLS%/convert_mnist_data.exe" %DATA%/t10k-images-idx3-ubyte %DATA%/t10k-labels-idx1-ubyte mnist_test_%BACKEND% --backend=%BACKEND%
echo "Done."
pause
(3)运行测试
在caffe-windows根目录下新建一个run_mnist.bat,文件中代码:
.\Build\x64\Release\caffe.exe train --solver=examples/mnist/lenet_solver.prototxt
pause