发布时间:2023-02-28 文章分类:电脑基础 投稿人:樱花 字号: 默认 | | 超大 打印

文章目录

  • 安装WSL
  • 安装Docker
  • 安装Docker镜像
    • 下载Docker镜像
    • 启动
    • gpu启动
    • 传送文件
    • 训练yolov5

安装WSL

Windows10和11支持Docker的安装,安装需要用到WSL。所以,我们先安装WSL。
参考文章:旧版 WSL 的手动安装步骤
以管理员身份打开powershell, 执行以下命令:

dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart

安装 WSL 2 之前,必须启用“虚拟机平台”可选功能。 计算机需要虚拟化功能才能使用此功能。

以管理员身份打开 PowerShell 并运行:

dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart

下载 Linux 内核更新包:l链接:
https://wslstorestorage.blob.core.windows.net/wslblob/wsl_update_x64.msi
下载完成后更新。

将 WSL 2 设置为默认版本,命令:

wsl --set-default-version 2

打开 Microsoft Store,搜索WSL,选择你偏好的 Linux版本。
Docker在Windows环境的搭建和使用
我选择的是Ubuntu22.04版本。点进去就可以看到下载链接Docker在Windows环境的搭建和使用
下载即可。下载完成后,启动系统,配置用户名和密码。
Docker在Windows环境的搭建和使用

安装Docker

下载链接:https://www.docker.com/get-started/,点击进去。

Docker在Windows环境的搭建和使用
点击红框的按钮,后可以看到下载链接。
Docker在Windows环境的搭建和使用
点击下载按钮,下载最新的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在Windows环境的搭建和使用
推荐使用这种方式。

传送文件

从Docker环境将文件复制到Win11系统

docker cp yolov5:/usr/src/app/runs/detect D:\datasets\uniform\runs

将Win11系统的文件传送到Docker环境中

docker cp D:\TrackAndReID\yolov5-master yolov5:/usr/src/app/

Docker在Windows环境的搭建和使用

训练yolov5

打开docker,选择运行的yolov5容器,单机红框的位置,进入容器的里面。
Docker在Windows环境的搭建和使用
选择Terminal就可以像在pycharm里面一样输入命令了。

Docker在Windows环境的搭建和使用
如果不喜欢还可以点击右上角,红框位置的Open in external terminal,重新开启一个窗口。

Docker在Windows环境的搭建和使用
然后输入命令:

cd yolov5-master
python train.py

开始训练yolov5
Docker在Windows环境的搭建和使用