系统支持情况

windwos + wsl(ubuntu 24.04.1 LTS)

安装 amd 显卡驱动

1
2
3
amdgpu-install -y --usecase=wsl,rocm --no-dkms
wget https://repo.radeon.com/amdgpu-install/6.3.4/ubuntu/noble/amdgpu-install_6.3.60304-1_all.de
sudo apt install ./amdgpu-install_6.3.60304-1_all.deb

查看 corm 安装情况

1
rocminfo

预期输出(显示对应显卡型号)

1
2
3
4
5
*******
Agent 2
*******
Name: gfx1201
Marketing Name: AMD Radeon RX 9070

安装 venv 库并创建虚拟环境

1
2
3
sudo apt install python3.12-venv
python3 -m venv venv
source venv/bin/activate

安装 pytorch

1
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.3

安装 yolo 环境

1
pip install ultralytics