windows10配置caffe-gpu

配置完windows-caffe-gpu,记录一下过程。希望对以后工作有帮助。

环境:
系统:Windows10 64位
GPU:Quadro p4000
cuda: 8.0
cudnn: 5.0
框架: caffe
语言: python(anaconda2)
编译软件:vs2013

一,安装驱动

1,下载驱动:驱动下载地址
在这里插入图片描述

 建议:1,语言项最好选择English(US)

2,安装
windows下显卡驱动安装,只需要默认安装,一直点下一步即可。安装完成后需要重启电脑,让驱动生效。

二,安装VS2013

1,下载安装
官网下载速度比较慢,同时需要密钥,推荐一位大佬的博客。
注意:下载64位的vs2013,建议下载专业版。

安装后,打开vs2013,点击“帮助”-“注册产品”,输入上面链接中对应的注册码。当出现如图所示情况即为破解成功。
在这里插入图片描述

2,添加环境变量
VS的安装路径下有一个VC文件夹,将其中的bin目录添加值环境变量(该目录下有cl.exe),路径参考如下:(以默认安装路径为例)

C:\Program Files (x86)\Microsoft Visual Studio 12.0\VC\bin

三,安装anaconda2

1,下载

下载地址

在这里插入图片描述

2,安装
建议:1,在C盘根目录下新建一个anaconda2文件夹
2,在安装过程中记得勾选添加anaconda2环境变量
3,其他默认安装即可

3.opencv环境
把cv2.pyd文件放入C:\Anaconda2\Lib\site-packages文件夹中

cv2.pyd下载地址:

https://download.csdn.net/download/amanfromearth/10395218

四,安装CUDA8.0

1,官网下载:

https://developer.nvidia.com/cuda-80-download-archive

官网下载速度很慢,在此推荐一个下载工具PanDownload。
在此分享我下载的百度云cuda8.0文件。

https://pan.baidu.com/s/1NTaFxrC36MBge2qLT5aMHQ
提取码:fhgw

2.安装
一直默认安装即可,安装结束后记得重启电脑。

五,安装CUDNN

1,下载文件
前面参考的文章,安装的是cudnn5.1,但是在安装过程中,我的电脑出现了一些问题,尝试安装cudnn5.0后得以解决。
cudnn5.1与cudnn5.0在官网很难下载,其他地方也需要会员下载,为方便各位,在此直接分享我的百度云文件。
cudnn5.1文件

https://pan.baidu.com/s/1HbNh4Smj2vHDVyPX-E40Kw
提取码:xszb

cudnn5.0文件

https://pan.baidu.com/s/1VBC2Cc8GqAmgxAwNCu693Q
提取码:paav

2.解压
先解压,稍后再用。

六,caffe配置

1.下载
1.caffe下载地址:https://github.com/Microsoft/caffe
2.NugetPackages下载:

https://pan.baidu.com/s/1by2esTnbejluHKtPlEq3ZA
提取码:sxhh

NugetPackages下载完后放在与caffe-master同级文件夹下
在这里插入图片描述

2.解压
解压到C:\caffe-master,把cudnn解压得到的cuda文件夹拷贝到caffe-master文件夹中。
3,配置
(1)打开C:\caffe-maste\windows文件夹,将其中CommonSettings.props.example文件复制一份,重命名为CommonSettings.props

(2)打开CommonSettings.props
a. 把其中的CudaVersion改为8.0
b. 把PythonSupport改为true
c. 把CuDnnPath改为刚才拷贝cuda文件夹的地址,即C:\caffe-master
d. 将CudaArchitecture改成compute_61,sm_61(这里的61是根据1080的计算力得到 的,其他显卡的计算力在这里查询:https://developer.nvidia.com/cuda-gpus
e. 把PythonDir改为C:\Anaconda2

(3)打开C:\caffe-master\windows路径下的Caffe.sln,添加faster rcnn需要的roi pooling laye
a.在libcaffe项目下的 cu/layers 文件夹右击,添加——现有项,找到caffe-master\src\caffe\layers下的roi_pooling_layer.cu文件,点击添加
b.在libcaffe项目下的 src/layers 文件夹右击,添加——现有项,找到caffe-master\src\caffe\layers下的roi_polling_layer.cpp文件,点击添加
c.在libcaffe项目下的 include/layers 文件夹右击,添加——现有项,找到caffe-master\include\caffe\layers下的roi_polling_layer.hpp文件,点击添加

4.编译libcaffe
(1)选中libcaffe,点击菜单栏的调试-libcaffe属性
(2)选择配置属性-常规,把配置改为Releas,平台改为活动(x64)
(3)选择C/C++,把将警告视为错误改为否
(4)点击确定关闭窗口。右键选择libcaffe,点击生成(会下载依赖项,所以需要网络)
(5)成功之后,在caffe-master文件夹中就会有一个Build文件夹。

5.编译其他
(1)右键解决方案,属性
(2)选择通用属性-启动项目-当前选定内容并确定
(3)检查本地windows调试器,调整为Release
(4) 右键解决方案,点击生成解决方案
(5)等待编译成功。

6.移动文件
(1)将caffe-master\Build\x64\Release\pycaffe\caffe文件夹移动到C:\Anaconda2\Lib\site-packages文件夹中
(2)打开cmd,输入python,再输入import caffe,没有报错的话,caffe就此配置成功。
如图:
在这里插入图片描述

  • 2
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小俊俊的博客

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值