Win10+Caffe&CPU&GPU+VS2013安装记录

2 篇文章 0 订阅
2 篇文章 0 订阅

刚刚解决了CUDA的安装问题,接下来就动手编译试试:

Caffe介绍(百度百科)

Caffe:全称Convolutional Architecture for Fast Feature Embedding。是一种常用的深度学习框架,主要应用在视频、图像处理方面的应用上。

需要准备的东西

点击标题进入对应网站下载
1. VS2013
2. CUDA 我用的是v9.2 (CPU Only不需要)
3. OpenCV 我用的是3.4.2 (CPU Only不需要)
4. CuDNN Library (CPU Only不需要)
5. Python(可选) 我安装了3.7 debug需要vs2015以上
6. NugetPackages 首次编译的时候会自动下载,会造成VS假死现象

开始动手

  1. 安装VS2013(推荐使用,其他版本编译可能会报其他错误)
    查看CUDA是否装好
    装好CUDA并且培植成功的话再VS中会多出一个菜单项
  2. 安装CUDA,NVDIA Developer可以下载,需要登录NVDIA账户。可以参考CUDA编程(一):背景与安装。如果安装失败,可以参考Win10 NVDIA CUDA安装失败解决方案CUDA安装程序会自动配置环境变量。
  3. Clone Caffe SSD Windows版本 Github传送门
  4. 进入clone保存的文件夹:
    \caffe-ssd-microsoft-master\windows,复制一份CommonSettings.props.example并重命名为CommonSettings.props
  5. 打开工程,加载CommonSettings.props
    打开工程

修改配置文件

善用Ctrl+F快速查找
其实打开CommonSettings.props就能看到官方其实已经写好了注释,我们根据提示即可。
需要修改的地方:

CPU Only
<CpuOnlyBuild>true</CpuOnlyBuild>
<UseCuDNN>false</UseCuDNN>

如果需要Python、MATLAB之类的支持可以在这些标记中改成true启用

//Example:启用Python支持
<PythonSupport>true</PythonSupport> // 是否启用python支持
<PythonDir>F:\Program Files\Python</PythonDir>//Python安装路径,可在系统变量PATH中查看

*然后下面的CuDNN啥的都不用管了

With GPU
<!--NOTE: CpuOnlyBuild and UseCuDNN flags can't be set at the same time.-->
<CpuOnlyBuild>false</CpuOnlyBuild>
<UseCuDNN>true</UseCuDNN>
<CudaVersion>9.2</CudaVersion>

然后填入CuDNN路径

<CudaVersion>9.2</CudaVersion> //CUDA 版本号,我自己安装的是9.2
<UseCuDNN>true</UseCuDNN> // true 代表使用cuDNN库
<!-- CuDNN 3 and 4 are supported -->
<CuDnnPath>这里填写CuDnn的路径</CuDnnPath>

cudnn路径填到解压出来的cuda的上一级路径即可。

即路径下是包含有cuda文件夹那一层,而非cuda文件夹里面那层,例如F:\Deep Learning\cudnn

修改caffe项目属性


警告不报错
更改为Release 平台选择 x64

生成libcaffe

生成libcaffe
使用CPU Only编译完美通过,没有爆错,但使用GPU就爆错了,如图:
GPU编译爆错

GPU编译error MSB3721错误解决方案

打开 cudnn.hpp ,找到

int pad_h, int pad_w, int stride_h, int stride_w) {
  CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
      pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION));
}

修改为:

int pad_h, int pad_w, int stride_h, int stride_w) {
  CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
      pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION,CUDNN_DATA_FLOAT));
}

修改
其实就是多加了一个CUDNN_DATA_FLOAT
重新生成 libcaffe
GPU libcaffe编译成功
GPU libcaffe编译成功。
然后我们把整个项目生成一下:
GPU整个项目完成
大功告成!
生成的文件存放在\Build\x64\Release
GPU生成的文件


后记(踩坑总结)

  1. 安装CUDA之前先把之前已经安装的NVDIA驱动卸载了
  2. 注意CuDNN解压路径
  3. 编译GPU版本的时候记得加上CUDNN_DATA_FLOAT否则一大把错误。(这个问题应该会在日后的版本中得到修复)
  4. 开启Python支持容易爆错,目前未找到解决方法….

附录

[NugetPackages 下载地址]
百度云
和git下来的文件夹放在同一级这里写图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在Ubuntu 16.04中安装Docker并使用GPU时出现问题,首先需要确保你的系统配置符合Docker和Caffe GPU的要求。以下是一些可能的解决方案: 1. 确保你的系统支持NVIDIA GPU:首先,检查你的计算机上是否安装了NVIDIA显卡驱动程序。可以使用以下命令检查驱动程序是否已成功安装并正在运行: ``` nvidia-smi ``` 如果该命令能够成功启动,并显示NVIDIA显卡的信息,则说明驱动程序已成功安装。 2. 安装NVIDIA Docker:Docker在GPU支持方面需要使用NVIDIA Docker插件。你可以按照NVIDIA Docker官方文档中的步骤安装,包括添加apt存储库、安装软件包和配置插件等。 3. 配置NVIDIA运行时:在使用Docker时,使用NVIDIA运行时可以确保Docker能够正确地使用GPU。你可以通过编辑`/etc/docker/daemon.json`文件来配置NVIDIA运行时,确保该文件中包含以下内容: ``` { "runtimes": { "nvidia": { "path": "nvidia-container-runtime", "runtimeArgs": [] } }, "default-runtime": "nvidia" } ``` 保存并退出该文件,然后重新启动Docker服务: ``` sudo systemctl restart docker ``` 4. 运行Docker容器使用GPU:在运行Caffe容器时,确保你在`docker run`命令中使用了`--runtime=nvidia`选项,以确保容器能够使用GPU。例如: ``` docker run --runtime=nvidia caffe-gpu-image ``` 这将使用NVIDIA运行时在容器中启动Caffe,并允许它使用GPU加速。 通过执行以上步骤,你应该能够在Ubuntu 16.04上安装Docker并成功使用GPU运行Caffe。如果问题仍然存在,请检查是否遵循了正确的安装步骤,并查看系统日志以获取更多信息。如果问题仍然无法解决,你可以搜索相关的错误信息或向社区等其他资源寻求帮助。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值