最近心血來潮想學一下YOLOv3,因而就去網上看了YOLOv3在win10下的配置教程。在配置過程當中塌坑無數,花了不少時間和精力,因此我想就此寫一篇博客來介紹在在win10+vs2015的環境下如何配置YOLOv3。git
儘可能安裝vs2015github
2.1肯定本身電腦對應的CUDA版本號shell
2.1.1打開控制面板windows
2.1.2進入NVIDIA控制面板工具
2.1.3選擇系統信息
測試
2.1.4點擊組件,便可看到本身電腦適應的CUDA的版本號ui
2.2進入官網https://developer.nvidia.com/cuda-downloads ,按照本身的系統選擇合適的版本。spa
CUDNN的版本須要與CUDA版本對應,下載的時候根據選定的CUDA版本進行選擇便可。.net
下載連接: 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
複製完以後要去檢查一下cuda和cudnn的環境變量 ,務必要保證這些環境變量都有要否則最後在測試時會報錯
下載地址: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配置到環境變量
下載地址: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時,那麼就恭喜你配置成功過了!
6.1下載yolov3.weights
官網:https://pjreddie.com/media/files/yolov3.weights
下載速度有點慢
也能夠用個人連接 連接: https://pan.baidu.com/s/1FSkXMGqCeqDFpeDxJzb3sA 提取碼: 1ajt
下載完成後把它放到darknet\build\darknet\x64下
6.2測試
在darknet\build\darknet\x6下按住shift鍵,點擊鼠標右鍵選擇「在此處打開Powershell 窗口(s)」
在命令行輸入 .\darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
若是看到如下結果,說明你成功的配好了YOLOv3