Caffe是目前非常流行的深度学习框架,使用C++/CUDA编写,使用方便,性能优异,适合线上环境部署。
原生Caffe是在Linux下编译部署的。对于初学者而言,大量依赖包需要花非常大代价才能编译成功,让人望而生畏。况且在Linux下阅读代码是一件非常头疼或蛋疼的事情。
为此,研究在Windows下搭建Caffe开发环境是利国利民的好事。微软不负众望在Github上开源了Microsoft版Caffe,链接:Microsoft Caffe
只有代码还不够,需要准备一台机器。
如果手头没有机器而且也不想攒机,可以考虑买一台阿里云HPC(购买链接:阿里云HPC),配备双 Tesla K40m 或 双 Tesla M40,专为深度学习定制机型。
购买后默认操作系统是Cent OS 7,通过提交工单可以重置为Windows 系统。为了完成本文实验,推荐重置为 Windows Server 2012 R2 64bit版。
编译环境:Visual Studio 2013 Ultimate版, 获取地址:http://download.microsoft.com/download/9/3/E/93EA27FF-DB02-4822-8771-DCA0238957E9/vs2013.5_ult_chs.iso?type=ISO
CUDA Toolkit版本:7.5,获取地址:https://developer.nvidia.com/cuda-downloads
CUDNN版本:4.0, 获取地址:https://developer.nvidia.com/cudnn
依次安装GPU驱动、Visual Studio 2013、CUDA Toolkit。将Microsoft/caffe代码下载到本地磁盘,本文路径为C:\Users\Administrator\Desktop\caffe-master。
将CUDNN解压到C:\Users\Administrator\Desktop\cuda。准备工作完毕。
进入C:\Users\Administrator\Desktop\caffe-master\windows目录,将文件CommonSettings.props.example
重命名为CommonSettings.props,修改其内容如下: