配置完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就此配置成功。
如图: