ROCm（仅支持Linux）

一、ROCm 简介

ROCm（Radeon Open Compute）是 AMD 推出的开源 GPU 计算平台，旨在为高性能计算（HPC）和人工智能（AI）工作负载提供支持。它类似于 NVIDIA 的 CUDA 平台，但专为 AMD GPU 设计。

ROCm 的主要组件包括：

• HIP（Heterogeneous-computing Interface for Portability）：一种用于编写可在 AMD 和 NVIDIA GPU 上运行的代码的编程模型。
• ROCm 驱动程序：用于管理 AMD GPU 的硬件资源。
• ROCm 编译器和运行时库：支持将高层代码编译并运行在 GPU 上。
• 开发工具：包括调试器、性能分析器等。

ROCm 的目标是提供一个开放、可移植且高性能的 GPU 计算平台，支持多种编程语言和应用场景。

二、为什么要安装 ROCm？

随着深度学习和科学计算的广泛应用，GPU 的计算能力变得尤为重要。ROCm 提供了在 AMD GPU 上运行这些计算任务的能力，使得开发者可以利用 AMD 显卡的性能优势。对于拥有 AMD Radeon RX 9070XT 显卡的用户，安装 ROCm 可以：

• 利用 GPU 加速深度学习模型的训练和推理。
• 在本地环境中运行高性能计算任务。
• 避免依赖昂贵的 NVIDIA GPU，降低成本。

三、Ubantu22安装 ROCm

ROCm 主要为 Linux 系统设计。

安装 ROCm

官网： https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/quick-start.html
在 Ubuntu22 终端中，执行以下命令安装 ROCm：

wget https://repo.radeon.com/amdgpu-install/6.4.1/ubuntu/jammy/amdgpu-install_6.4.60401-1_all.deb
sudo apt install ./amdgpu-install_6.4.60401-1_all.deb
sudo apt update
sudo apt install python3-setuptools python3-wheel
sudo usermod -a -G render,video $LOGNAME # Add the current user to the render and video groups
sudo apt install rocm

#安装AMDGPU驱动
wget https://repo.radeon.com/amdgpu-install/6.4.1/ubuntu/jammy/amdgpu-install_6.4.60401-1_all.deb
sudo apt install ./amdgpu-install_6.4.60401-1_all.deb
sudo apt update
sudo apt install "linux-headers-$(uname -r)" "linux-modules-extra-$(uname -r)"
sudo apt install amdgpu-dkms

#验证安装
rocminfo

4. 安装 PyTorch（可选）

如果使用 PyTorch 进行深度学习任务，可以安装支持 ROCm 的 PyTorch 版本：

sudo apt install python3-pip -y
pip3 install --upgrade pip wheel
#安装 PyTorch（ROCm 版本）
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm6.1.3