部署YOLOV5環(huán)境到服務(wù)器進行多GPU訓(xùn)練
# 部署YOLOV5環(huán)境到服務(wù)器進行多GPU訓(xùn)練
### 訪問官網(wǎng)進行注冊并登錄
[點我訪問AutoDl官網(wǎng)](https://www.autodl.com/home)
#### 購買一臺GPU服務(wù)器:
境到服務(wù)器進行多GPU訓(xùn)練/1.png)
### 通過vscode鏈接GPU服務(wù)器:
#### 安裝插件
境到服務(wù)器進行多GPU訓(xùn)練/2.png)
#### 配置連接信息
境到服務(wù)器進行多GPU訓(xùn)練/3.gif)
### 安裝服務(wù)器中的YOLOV5-GPU環(huán)境
#### 到torch官網(wǎng):
[點我訪問torch官網(wǎng)](https://pytorch.org/get-started/locally/)
境到服務(wù)器進行多GPU訓(xùn)練/4.png)
#### 復(fù)制命令到終端,安裝pytorch,記得把pip3改為pip
```
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
```
境到服務(wù)器進行多GPU訓(xùn)練/5.png)
#### 最后等待安裝完成即可
### 復(fù)制自己本機跑通的YOLOV5代碼到服務(wù)器
境到服務(wù)器進行多GPU訓(xùn)練/6.gif)
#### 注釋這里我們通過vscode進行上傳文件,也可以通過其他方式進行上傳
#### 先解壓文件
境到服務(wù)器進行多GPU訓(xùn)練/7.png)
#### 進入yolov5-7.0文件夾根目錄將tqdm>=4.64.0改為tqdm>=4.63.1輸入以下命令:
```
pip install -r requirements.txt
```
### 在服務(wù)器中安裝python擴展
境到服務(wù)器進行多GPU訓(xùn)練/8.png)
### 重連服務(wù)器,重新打開目錄yolov5目錄準(zhǔn)備訓(xùn)練
境到服務(wù)器進行多GPU訓(xùn)練/9.gif)
#### 如果出現(xiàn)Downloading https://ultralytics.com/assets/Arial.ttf to /root/.config/Ultralytics/Arial.ttf...:
```
多等一會或者多運行幾次即可,實在不行可以手動在本地找到這個字體文件復(fù)制到服務(wù)器中的/root/.config/Ultralytics/目錄下
```
### 因為代碼我們在本地已經(jīng)跑通了,所以直接運行train.py即可進行默認(rèn)單GPU訓(xùn)練:
境到服務(wù)器進行多GPU訓(xùn)練/10.gif)
### 如果你需要進行多GPU訓(xùn)練,可以參考以下步驟,前提是你真的買了兩個GPU哦:
境到服務(wù)器進行多GPU訓(xùn)練/11.png)
#### 既然是多GPU就要合理調(diào)試batch-size中的值,可以適當(dāng)微調(diào),觀察顯卡顯存,如果顯存沒滿可以持續(xù)增加batch-size,默認(rèn)訓(xùn)練epochs的值是100,如果你數(shù)據(jù)足夠多,可以搞500epochs,當(dāng)然了多GPU訓(xùn)練請先確保你的數(shù)據(jù)有5000-30000數(shù)據(jù)再使用。
### 好了本期教學(xué)分享到這里