基于Ubuntu22.04.1的ROCm环境搭建

基于 ubuntu22.04.1 搭建 ROCm-5.6.0 环境

1. 安装 gnupg2

sudo apt update
sudo apt install gnupg2

2. 获取安装脚本并安装

wget https://repo.radeon.com/amdgpu-install/5.6/ubuntu/jammy/amdgpu-install_5.6.50600-1_all.deb
sudo apt install ./amdgpu-install_5.6.50600-1_all.deb

3. 安装单个用例

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

4. 重启计算机并验证

sudo reboot
重启之后使用 ROCm 管理工具验证：rocm-smi
该命令会显示出显卡温度、占用率等信息，至此 ROCm 环境搭建完成