Windows 上 Caffe开发环境搭建

  1. 云栖社区>
  2. 博客>
  3. 正文

Windows 上 Caffe开发环境搭建

以亭 2016-03-28 15:21:37 浏览6686
展开阅读全文

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,修改其内容如下:

网友评论

登录后评论
0/500
评论
以亭
+ 关注