
1.Caffe的主版本只支持Linux,所以要下载专门的Caffe Windows版本,网址为
https://github.com/niuzhiheng/caffe
该版本需要的环境如下
1)Windows 64-bit
2)MS Visual Studio 2012
3)CUDA toolkit 6.5
4)Other dependencies which you can directly download from here.
2.下载VS2012和CUDA
1)VS2012。
2)CUDA 6.5下载地址https://developer.nvidia.com/cuda-downloads。AMD显卡也可以直接安装,但是应该不能使用。
3.配置第三方库
可以直接在优快云上找到caffe运行依赖的打包文件,名字叫dependency-20140804.7z或者之类的都可以,解压到相应目录。
打开MainBuilder.sln工程文件,属性->C/C++->常规->附加包含目录添加头文件,属性->链接器->常规->附加包含目录添加lib文件目录
属性->链接器->输入 添加../../3rdparty/lib下的.lib文件名。
4.用VS2012打开\build\MSVC目录下的MainBuilder.sln工程文件,选择”x64”,进行编译。
编译时可能会出现std::stuple模板参数之类的问题,估计是VS2012的问题,在C/C++ ->预处理器 -> 预处理器定义添加 _VARIADIC_MAX=10 可解决问题。
编译后在./bin下生成MainCaller.exe文件。
测试时,可在./caffe-windows/example/mnist下点击get_mnist_leveldb.bat下载mnist手写字体数据库,下载后运行train_lenet.bat进行训练。
(用CPU训练的话,我的电脑大概训练了一个小时?)
使用GPU训练时,可用记事本打开lenet_solver.prototxt文件,将最后一行的GPU参数改为CPU即可。
由于暂时不使用GPU,GPU详细设置先暂告一段落。