系统支持情况
windwos + wsl(ubuntu 24.04.1 LTS)
安装 amd 显卡驱动
1 | amdgpu-install -y --usecase=wsl,rocm --no-dkms |
查看 corm 安装情况
1 | rocminfo |
预期输出(显示对应显卡型号)
1 | ******* |
安装 venv 库并创建虚拟环境
1 | sudo apt install python3.12-venv |
安装 pytorch
1 | pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.3 |
安装 yolo 环境
1 | pip install ultralytics |