AI – Yolo Model Training – Darknet Win10 – Environment Setting

簡介

經過之前透過Darkflow來訓練自己模型後,因為該訓練平台不需要安裝一些有的沒有的,但這次要使用的Darknet就是比較正規的訓練平台,且要使用GPU來加速訓練,所以這邊先列一下要安裝的軟體。

  • NVIDIA CUDA (版本可以裝10.1 ,但最新的是11.1)
  • NVIDIA cuDNN (v8.0.3)
  • OpenCV (v4.1)
  • Visual Studio 2017 (2019也可以)

我這邊先講一下安裝順序:

先安裝Visual Studio(之後要Build darknet.exe)必要工具→CUDA→cuDNN→Clone Darknet→OpenCV。

安裝與設定CUDA + cuDNN

下載:CUDA
下載:cuDNN

首先我們先按裝最重要的GPU訓練軟體,CUDA安裝時照理來說只需要一直下一步就好,cuDNN下載下來先解壓縮。

在安裝CUDA時軟體會分析你需要安裝那些東西,因為我電腦已經先前安裝好CUDA所以上圖所顯示的只須安裝Nsight,這個東西在Build darknet.exe必須要安裝的項目。

接下來把cnDNN的解壓縮檔案bin、include、lib複製到剛剛安裝的v11.1資料夾下。

安裝好CUDA,就要到電腦的環境設定設定環境變數,請到你安裝CUDA的安裝完成路徑,找到CUDA,把路徑都貼到Path中,新增→貼上,圖二是我貼上的相關路徑,保險一點就把所有路徑都貼上去。

假如上述都完成後,請大家到命令題是字元輸入下方指令,檢查一下是否安裝成功,假如成功就會顯示上圖資訊,會列出目前擬電腦的CUDA版本。

nvcc -V

下載DarkNet + OpenCV 4.1.0

我們先下載AlexeyAB提供的Window版本,下方是用命令題是字元clone,也可以直接下載zip檔案。

git clone https://github.com/AlexeyAB/darknet

Darknet先放著,我們先安裝也是蠻重要的軟體Opencv,安裝時他會把該軟體接壓縮在一個地方再進行安裝。

我們這邊先把解壓縮好的檔案剪下貼上在C槽,並設置OpenCV的環境變數,一樣到系統→進階系統設定→系統變數→Path,將上圖路徑貼在裡面。

重要的來了,請把解壓縮好的Opencv資料夾中的:opencv\build\bin兩個dll檔案複製貼上在:darknet\build\darknet中。這在之後Build darknet.exe也是必備檔案之一。

小結

這篇先講到這裡,以上步驟都必須準確且確實的執行到,這樣銜接下一篇章的Build darknet.exe才不會出錯,環境設定確實生硬了一點,但是也沒辦法,只好乖乖照做囉!

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *