如何在Arch Linux系统上正确地安装ROCm(AMD异构计算平台)?
时间: 2024-12-06 12:25:05 浏览: 53
在 Arch Linux 系统上安装 ROCm(AMD 的开源加速计算框架),可以按照以下步骤操作:
1. **更新包列表**:
```bash
sudo pacman -Syu
```
2. **添加存储仓库**:
添加 AMD 官方软件源到 `/etc/pacman.conf` 文件(如果尚未存在)。在 `[extra]` 或 `[multilib]` 区域添加:
```
[amdrocm]
Server = https://repo amd.com/rocm/ $arch
```
之后运行:
```bash
sudo pacman-key --populate amdrocm
sudo systemctl start yay.service
```
3. **安装 AUR 库**:
使用 `yay` 来安装 AUR 上的 ROCm 工具包。首先,需要安装 `yay`:
```bash
sudo pacman -Syyu yay
```
然后通过 `yay` 安装 ROCM:
```bash
yay -S rocm-<package-name>
```
具体替换 `<package-name>` 为你需要的 ROCm 组件,如 `rocm-cmakelists`, `rocm-dkms`, `rocm-opencl-dev` 等。
4. **安装驱动和 SDK**:
可能需要分别安装不同组件:
```bash
yay -S rocm-install rocm-dkms rocm-opencl-rocr runtime-libs rocm-utils
```
5. **设置环境变量**:
配置环境变量,使得 ROCm 工具可以在终端中使用。通常会在 `/etc/profile.d/rocm.sh` 或者用户 `.bashrc` 中设置:
```bash
export PATH="/opt/rocm/bin:$PATH"
export LD_LIBRARY_PATH="/opt/rocm/lib:$LD_LIBRARY_PATH"
```
6. **激活模块(如有需要)**:
如果你的 CPU 支持 ROCM,可能需要加载对应的内核模块,例如:
```bash
modprobe amdgpu
```
7. **检查安装**:
最后,你可以通过命令 `rocminfo` 检查 ROCm 是否安装成功,并确认驱动状态。
阅读全文