1. 环境准备:搭建RK3576开发环境
在开始YOLOv8模型的采集、标注、训练和部署之前,我们需要先搭建好开发环境。RK3576是瑞芯微推出的一款高性能AI芯片,支持多种深度学习框架的推理加速。我们将使用Windows系统下的WSL(Windows Subsystem for Linux)作为开发环境,通过VSCode的SSH远程连接功能进行开发。
1.1 安装和配置WSL
WSL让我们可以在Windows系统上运行Linux环境,这对于深度学习开发非常方便。以下是详细的安装步骤:
- 以管理员身份打开PowerShell,运行以下命令启用WSL功能:
bash复制dism.exe /online /enable-feature /featurename:Microsoft-Windows-Subsystem-Linux /all /norestart
dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart
-
下载并安装WSL2内核更新包(可从微软官网获取)
-
设置WSL2为默认版本:
bash复制wsl --set-default-version 2
- 从Microsoft Store安装Ubuntu 20.04 LTS(推荐版本)
注意:如果遇到"an error occurred while running a wsl command"错误,通常是由于虚拟化功能未开启或WSL安装不完整导致的。可以尝试在BIOS中启用虚拟化技术(VT-x/AMD-V),并确保Hyper-V功能已启用。
1.2 配置SSH远程连接
为了能够在VSCode中远程连接到WSL环境,我们需要配置SSH服务:
- 在WSL中安装并启动SSH服务:
bash复制sudo apt update && sudo apt install openssh-server
sudo service ssh start
- 修改SSH配置以允许密码登录:
bash复制sudo nano /etc/ssh/sshd_config
找到并修改以下参数:
code复制PasswordAuthentication yes
PermitRootLogin yes
- 重启SSH服务:
bash复制sudo service ssh restart
- 在VSCode中安装"Remote - SSH"扩展,然后添加新的SSH连接,地址为:
code复制ssh username@localhost -p 22
提示:如果遇到"ssh: could not resolve hostname"错误,请检查SSH服务是否正常运行,以及防火墙是否阻止了连接。
1.3 安装CUDA和cuDNN
由于YOLOv8训练需要GPU加速,我们需要在WSL中安装CUDA和cuDNN:
-
首先安装NVIDIA驱动(需要在Windows主机上安装)
-
在WSL中安装CUDA Toolkit 11.7(与RK3576兼容的版本):
bash复制wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/wsl-
