Ubuntu24.04(Linux) 部署支持 ROCm 7+ 的推理框架部署指南 — 环境准备部分
本节介绍如何在 Ubuntu 24.04 上:
- 清理已有的 ROCm / AMD 相关环境
- 使用官方脚本安装 ROCm 7.1.0
- 通过工具命令验证安装是否成功
sudo apt remove rocm*
sudo apt remove amd*如果使用 Docker,需要安装
amdgpu-dkms:参考
https://rocm.docs.amd.com/projects/install-on-linux/en/latest/how-to/docker.html
下方脚本已包含相关步骤;如果不执行脚本安装,需要自行手动安装。
下载安装脚本到本地:
https://github.com/amdjiahangpan/rocm-install-script/blob/ROCm_7.1.0_ubuntu_24.04/2-install.sh更新系统并执行安装脚本:
sudo apt update
# 更新内核
sudo apt upgrade -y
sudo bash 2-install.sh安装完成后,使用以下命令验证(均应有显卡相关输出):
rocminfo
rocm-smi
amd-smi更多安装细节可参考官方快速上手文档:
https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/quick-start.html