YOLOv3配置(win10+opencv3.40+cuda9.1+cudnn7.1+vs2015)
Posted songhouhou
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了YOLOv3配置(win10+opencv3.40+cuda9.1+cudnn7.1+vs2015)相关的知识,希望对你有一定的参考价值。
最近心血来潮想学一下YOLOv3,于是就去网上看了YOLOv3在win10下的配置教程。在配置过程中塌坑无数,花了很多时间和精力,所以我想就此写一篇博客来介绍在在win10+vs2015的环境下如何配置YOLOv3。
1.安装VS(YOLOv3支持vs2012、2013、2015,其他版本需要下载工具集)
尽量安装vs2015
2.安装CUDA
2.1确定自己电脑对应的CUDA版本号
2.1.1打开控制面板
2.1.2进入NVIDIA控制面板
2.1.3选择系统信息
2.1.4点击组件,即可看到自己电脑适应的CUDA的版本号
2.2进入官网https://developer.nvidia.com/cuda-downloads ,按照自己的系统选择合适的版本。
3.安装CUDNN
CUDNN的版本需要与CUDA版本对应,下载的时候根据选定的CUDA版本进行选择即可。
下载链接: https://developer.nvidia.com/cudnn
下载之后把对应文件夹的文件直接放入CUDA的对应文件夹就可以了
即
将D:\\安装包\\cudnn-9.1-windows10-x64-v7.1\\cuda\\bin\\cudnn64_7.dll 复制到C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v9.1\\bin
将 D:\\安装包\\cudnn-9.1-windows10-x64-v7.1\\cudacuda\\ include\\cudnn.h 复制到C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v9.1\\include
将D:\\安装包\\cudnn-9.1-windows10-x64-v7.1\\cudacuda\\lib\\x64\\cudnn.lib复制到C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v9.1\\lib\\x64
4. 安装 OpenCV (OpenCV 2.x.x 或者 OpenCV <= 3.4.0)
(切记opencv的版本不能超过3.40,我当时就是因为粗心没注意到这点才折腾了很久)
下载地址:https://sourceforge.net/projects/opencvlibrary/files/opencvwin/3.4.0/opencv-3.4.0-vc14_vc15.exe/download
如果访问不了,可以使用我的链接
链接: https://pan.baidu.com/s/1yACjz1AiDWtkUzxZHqCH0w 提取码: scxv
接下来将opencv配置到环境变量
5. 下载darknet
下载地址:https://github.com/AlexeyAB/darknet
(如果会使用git,则直接git clone https://github.com/AlexeyAB/darknet)
5.1修改darknet.vcxproj文件
进入darknet\\build\\darknet目录,用记事本打开darknet.vcxproj,将所有CUDA 10.0修改为自己对应的CUDA版本,本篇使用CUDA 9.1,故改为9.1。(共有两处需要修改)
5.2打开darknet.sln
5.2.1将项目修改为Release x64
5.2.2配置包含目录 + 库目录 + 链接器
包含目录:在darknet项目上点击鼠标右键->属性,弹出如下界面:然后VC++目录-->包含目录-->编辑
添加的目录:...\\opencv\\build\\include(...代表opencv的安装路径,下文同理)
...\\opencv\\build\\include\\opencv
...\\opencv\\build\\include\\opencv2
库目录:方法与包含目录类似,添加的目录为:...\\opencv\\build\\x64\\vc14\\lib
链接器:添加目录...\\opencv\\build\\x64\\vc14\\lib下库的名字:opencv_world340.lib
5.2.3右键darknet,选择属性,进入CUDA C/C++中的Code Generation,删掉compute_75,sm_75(此步极为关键,要不然必出错)
5.3右键darknet,选择生成
编译成功,会在darknet\\build\\darknet\\x64下得到darknet.exe
当你看到darknet.exe时,那么就恭喜你配置成功过了!
以上是关于YOLOv3配置(win10+opencv3.40+cuda9.1+cudnn7.1+vs2015)的主要内容,如果未能解决你的问题,请参考以下文章
win10下用yolov3训练WiderFace数据集来实现人脸检测(TensorFlow版本,darkface作为测试集)
win10下用yolov3训练WiderFace数据集来实现人脸检测(TensorFlow版本,darkface作为测试集)