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)