最新Ubuntu如何安装ROCm（AMD的机器学习驱动）

[yubinalice]

如题，

想使用YOLOv5跑个人流量测试，之前在论坛提出过要求建立机器学习新区，但是至今没被采纳，那就把YOLOv5分到影像区来发贴了。

一开始没安装AMD显卡的机器学习驱动，速度只有7fps,显然是不能接受的。

然后开始安装，流程如下：

sudo apt update

sudo apt install gnupg2

wget https://repo.randeon.com/amdgpu-install ... -1_all.deb
sudo apt install ./amdgpu-install_5.6.506000-1_all.deb

sudo amdgpu-install --usecase=rocm,hip,mllib --no-dkms

到这里，返回，

谁知道这个是怎么处理？

我用的是Ubuntu Studio 23.10。

[yubinalice]

另外，我使用的是AMD的核显Ryzen 3400G ，Vega 11核显，这个核显支持ROCm5.6？？

5.6是最后支持Vega显卡的。

[markawonge]

sudo apt update
wget https://repo.radeon.com/amdgpu-install/ ... -1_all.deb
sudo apt install ./amdgpu-install_5.6.50601-1_all.deb

sudo amdgpu-install --usecase=hiplibsdk,rocm

sudo usermod -aG video $USER
sudo usermod -aG render $USER

sudo reboot


23.10的kernel太新估计要折腾，ROCm5.6.1官方建议ubuntu 22.04.2(kernel 5.19)。