Windows下编译caffe

最新推荐文章于 2022-05-26 22:28:32 发布

鹊踏枝-码农

最新推荐文章于 2022-05-26 22:28:32 发布

阅读量458

点赞数

Deep Learning 同时被 3 个专栏收录

22 篇文章

订阅专栏

Win7

5 篇文章

订阅专栏

Caffe

5 篇文章

订阅专栏

本文详细介绍了在Windows环境下使用VS2013编译Caffe的过程，包括CPU模式和GPU模式的具体步骤，以及所需的CUDA 8.0和cuDNN v5.1等依赖配置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

来源：Windows下编译caffe

最近在windows上重新部署了下caffe，发现微软对提供的caffe做了很多改进，解决了很多编译配置的bug，程序下载caffe依赖包NugetPackages和编译速度也快了很多，现在上手caffe算是零门槛了。在win10 + vs2013 + caffe + cuda8.0 +cudnn v5.1 环境下编译成功。
1.cpu模式

1.1下载caffe-master

windows下使用caffe必须先安装VS编译器，VS2013具有很好的的兼容性，请预先安装VS2013。然后到github上下载微软提供的针对windows用户的caffe-master将解压放在某个文件夹下，例如桌面下。

1.2编译caffe

然后在caffe-master文件夹下的windows文件夹中，将CommonSettings.props.example文件复制并重命名为CommonSettings.props，再用记事本打开，作如下修改，表示目前使用cpu模式。

<CpuOnlyBuild>true</CpuOnlyBuild>
<UseCuDNN>false</UseCuDNN>
 
 1
2

然后用VS2013打开windows文件夹下的caffe.sln,打开后会发现整个解决方案总共包括了以下16个项目，其中项目libcaffe为整个解决方案的单启动项目，右键libcaffe项目->生成，程序会开始下载微软为windows用户提供的caffe依赖库–NugetPackages工具包，之后会出现没有生成“object”文件的错误，双击error弹出的提示框选择确定，然后关闭alt_sstream_impl.hpp并保存更改。重新编译libcaffe，右键libcaffe项目->重新生成，编译成功！然后对其它项目进行编译。
这里写图片描述

这里写图片描述

2.gpu模式（win10 + caffe + cuda8.0 + cudnn v5.1）

2.1安装NVIDIA显卡驱动

要使用gpu加速，首先你的显卡得是NVIDIA的。博主电脑win10，单显卡NVIDIA Geforce GTX 1060 6G，买来时已经预装了显卡驱动，如果你的电脑没有自带NVIDIA显卡驱动，需要自己到英伟达官网下载自己显卡和操作系统对应的驱动程序

这里写图片描述

2.2安装cuda8.0

可能是显卡比较新，下载了cuda7.5在安装开始检测兼容性不过关，而安装最新的cuda8.0检测兼容性通过，那索性就装最新的呗，安装路径可以就用默认的，cuda在安装的时候会自动关联到你的VS编译器 cuda8.0下载

这里写图片描述

2.3下载cudnn v5.1

目前windows已经支持cudnn v5.1，最新的caffe编译gpu版本时如果使用cudnn v4或cudnn v5编译会报关于NugetPackages中的opencv相关的错误，而cudnn v5.1则解决了这个bug，因此请用v5.1。到英伟达官网下载，选择cuDNN v5.1 Library for Windows 10 cudnn下载需要邮箱注册一下才能下载，下载解压到任意目录即可，里面是一个cuda文件夹，记住这个cuda文件夹所在的目录路径，之后要用到，例如我解压得到的cuda文件夹是在D:\caffe文件夹下。

这里写图片描述

2.4编译caffe

Caffe-master文件夹下windows目录中，将CommonSettings.props.example文件重命名为CommonSettings.props，再用记事本打开，作如下修改：

<CpuOnlyBuild>false</CpuOnlyBuild>
<UseCuDNN>true</UseCuDNN>
<CudaVersion>8.0</CudaVersion>
 
 1
2
3

其中cuda版本号改为我们安装了8.0版本，并设置cudnn路径为你刚刚解压得到的cuda文件夹所在的目录路径。

<!-- CuDNN 4 and 5 are supported -->
<CuDnnPath>D:\caffe</CuDnnPath>
 
 1
2

保存配置文件后重新打开caffe.sln，重新编译libcaffe项目，右键libcaffe->重新生成，编译成功！再对其它项目进行编译。不过好像微软提供的caffe还是不够稳定，有时候还是会报关于NugetPackages的opencv的错（明明刚刚都编译过了，过了会儿重新编译就报这个错），我一般是右键libcaffe清除，然后删了NugetPackages，再右键libcaffe生成，解决那个找不到object问题然后再重新生成就好了。

2.5编译matlab接口

可以参照博客微软caffe之matlab接口配置

关于python接口，以后慢慢更。现在不仅可以在linux上跑caffe了，windows上体验也很好。