目錄
- 安裝WSL
- 安裝Docker鏡像
- 下載Docker鏡像
- 啟動
- gpu啟動
- 傳送文件
- 訓練yolov5
- 開始訓練yolov5
- 更換python版本
- 添加PPA源
- 更新
- 安裝python3.8
- 修改python命令默認指向
- 查看默認python指向:
- 刪除原有python軟連接
- 建立python到python3.8新的軟鏈接
安裝WSL
Windows10和11支持Docker的安裝,安裝需要用到WSL。所以,我們先安裝WSL。
參考文章:詳解Windows 利用 WSL2 安裝 Docker 的2種方式
以管理員身份打開powershell, 執(zhí)行以下命令:
dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart
安裝 WSL 2 之前,必須啟用“虛擬機平臺”可選功能。 計算機需要虛擬化功能才能使用此功能。
以管理員身份打開 PowerShell 并運行:
dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart
下載 Linux 內(nèi)核更新包:l鏈接:
https://wslstorestorage.blob.core.windows.net/wslblob/wsl_update_x64.msi
下載完成后更新。
將 WSL 2 設置為默認版本,命令:
wsl --set-default-version 2
打開 Microsoft Store,搜索WSL,選擇你偏好的 Linux版本。
我選擇的是Ubuntu22.04版本。點進去就可以看到下載鏈接
下載即可。下載完成后,啟動系統(tǒng),配置用戶名和密碼。
安裝Docker
下載鏈接:https://www.docker.com/get-started/,點擊進去。
點擊紅框的按鈕,后可以看到下載鏈接。
點擊下載按鈕,下載最新的Docker。
雙擊安裝文件,一步步安裝。
安裝Docker鏡像
下載Docker鏡像
docker pull ultralytics/yolov5:latest
啟動
docker run -v yolov5 --ipc=host -it ultralytics/yolov5:latest
如果電腦沒有gpu,可以使用這種方式。
gpu啟動
格式如下:
docker run -itd --gpus all --name 容器名 -e NVIDIA_DRIVER_CAPABILITIES=compute,utility -e NVIDIA_VISIBLE_DEVICES=all 鏡像名
例如:
docker run -itd --gpus all --name yolov5 -e NVIDIA_DRIVER_CAPABILITIES=compute,utility -e NVIDIA_VISIBLE_DEVICES=all ultralytics/yolov5:latest
推薦使用這種方式。
傳送文件
從Docker環(huán)境將文件復制到Win11系統(tǒng)
docker cp yolov5:/usr/src/app/runs/detect D:\datasets\uniform\runs
將Win11系統(tǒng)的文件傳送到Docker環(huán)境中
docker cp D:\TrackAndReID\yolov5-master yolov5:/usr/src/app/
訓練yolov5
打開docker,選擇運行的yolov5容器,單機紅框的位置,進入容器的里面。
選擇Terminal就可以像在pycharm里面一樣輸入命令了。
如果不喜歡還可以點擊右上角,紅框位置的Open in external terminal
,重新開啟一個窗口。
然后輸入命令:
cd yolov5-master python train.py
開始訓練yolov5
這篇文章告訴大家如何在Docker環(huán)境中更換python版本。
更換python版本
添加PPA源
執(zhí)行命令:
sudo add-apt-repository ppa:deadsnakes/ppa
如果出現(xiàn):add-apt-repository: command not found
的問題,則執(zhí)行:
sudo apt-get update sudo apt-get install software-properties-common
add-apt-repository 命令是software-properties-common包的一部分,因此安裝這個包就OK了
更新
sudo apt update
安裝python3.8
sudo apt install python3.8 python3.8-dev -y
修改python命令默認指向
查看默認python指向:
ls -l /usr/bin | grep python
刪除原有python軟連接
rm /usr/bin/python
建立python到python3.8新的軟鏈接
ln -s /usr/bin/python3.8 /usr/bin/python