Rocky Linux安装rocm
时间: 2025-01-11 11:39:04 浏览: 175
### 如何在 Rocky Linux 上安装 ROCm
#### 准备工作
为了确保ROCm能够在Rocky Linux上顺利运行,需要确认操作系统内核版本以及硬件支持情况。通常情况下,AMD Radeon Instinct系列显卡和其他兼容GPU是必要的。
#### 更新系统包管理器缓存并升级现有软件包
建议先更新系统的软件源列表,并将已有的程序升级至最新稳定版:
```bash
sudo dnf update -y
```
#### 添加ROCm仓库
通过官方文档获取最新的YUM库链接地址,将其添加到当前使用的发行版中以便后续能够方便地下载所需组件。对于Rocky Linux来说,可以执行如下命令来设置相应的存储库:
```bash
sudo rpm --import https://repo.radeon.com/rocm/rocm.gpg.key
echo "[ROCm]" | sudo tee /etc/yum.repos.d/rocm.repo
echo "name=ROCm" | sudo tee -a /etc/yum.repos.d/rocm.repo
echo "baseurl=https://repo.radeon.com/rocm/yum/rpm/" | sudo tee -a /etc/yum.repos.d/rocm.repo
echo "enabled=1" | sudo tee -a /etc/yum.repos.d/rocm.repo
echo "gpgcheck=1" | sudo tee -a /etc/yum.repos.d/rocm.repo
echo "gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key" | sudo tee -a /etc/yum.repos.d/rocm.repo
```
#### 安装依赖项
某些特定工具和服务可能对ROCm的成功部署至关重要,因此要提前准备好这些前置条件:
```bash
sudo dnf install kernel-devel dkms elfutils-libelf-devel numactl numactl-devel hwloc libpciaccess-utils git wget vim cmake3 -y
```
#### 安装ROCm驱动及相关组件
现在可以从之前配置好的yum源里直接拉取ROCm及其关联模块了:
```bash
sudo dnf install rocm-dkms hipblas hipsparse miopen-hip rocfft rccl -y
```
#### 配置环境变量
为了让编译后的应用程序能找到对应的头文件路径和动态连接库位置,在~/.bashrc或者其他shell初始化脚本里面追加几条export语句:
```bash
echo 'export PATH=$PATH:/opt/rocm/bin' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/opt/rocm/lib:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
```
#### 测试安装成果
最后一步就是验证整个过程是否顺利完成,可以通过简单的例子来进行检验,比如跑通一个HIP Hello World实例[^1]。
阅读全文
相关推荐

















