当前位置: 首页 > news >正文

Fedora 43 配置 ROCm 7.2 与 PyTorch 2.11

Fedora 43 配置 ROCm 7.2 与 PyTorch 2.11

一、前言

自从 AMD ROCm 7.0 以上开始支持大量 Radeon 7000 系列显卡,博主准备将为自己的 Fedora 43 主机安装 ROCm 使自己的 7700XT 具有加速深度学习的能力,并且兼容 CUDA 可以轻松将现有基于 CUDA 开发的代码迁移至 ROCm 平台。

但是截止至 2026 年 4 月,AMD 官方并没有发布 Fedora 部署 ROCm 教程,不过 Fedora 社区一直在积极推进 ROCm 的适配,在博主搭载着 7700XT 的 Fedora 43 主机能够成功部署 ROCm 7.2 和 PyTorch 2.11. 并且能够顺利加速深度学习项目。

本文环境如下所示:

             .',;::::;,'.                 gudi@fedora.';:cccccccccccc:;,.             -----------.;cccccccccccccccccccccc;.          OS: Fedora Linux 43 (KDE Plasma Desktop Edition) x86_64.:cccccccccccccccccccccccccc:.        Kernel: Linux 6.19.12-200.fc43.x86_64.;ccccccccccccc;.:dddl:.;ccccccc;.      Uptime: 6 hours, 4 mins.:ccccccccccccc;OWMKOOXMWd;ccccccc:.     Packages: 2918 (rpm), 6 (flatpak)
.:ccccccccccccc;KMMc;cc;xMMc;ccccccc:.    Shell: bash 5.3.0
,cccccccccccccc;MMM.;cc;;WW:;cccccccc,    Display (default): 3440x1440 in 39"
:cccccccccccccc;MMM.;cccccccccccccccc:    DE: KDE Plasma 6.6.4
:ccccccc;oxOOOo;MMM000k.;cccccccccccc:    WM: KWin (Moba/X)
cccccc;0MMKxdd:;MMMkddc.;cccccccccccc;    WM Theme: Breeze
ccccc;XMO';cccc;MMM.;cccccccccccccccc'    Theme: Breeze (Light) [Qt], Breeze [GTK2/3]
ccccc;MMo;ccccc;MMW.;ccccccccccccccc;     Icons: Breeze [Qt], breeze [GTK2/3/4]
ccccc;0MNc.ccc.xMMd;ccccccccccccccc;      Font: Noto Sans (10pt) [Qt], Noto Sans (10pt) [GTK2/3/4]
cccccc;dNMWXXXWM0:;cccccccccccccc:,       Terminal: /dev/pts/0
cccccccc;.:odl:.;cccccccccccccc:,.        CPU: AMD Ryzen 7 5700X (16) @ 3.82 GHz
ccccccccccccccccccccccccccccc:'.          GPU: AMD Radeon RX 7700 XT [Discrete]
:ccccccccccccccccccccccc:;,..             Memory: 5.14 GiB / 31.22 GiB (16%)':cccccccccccccccc::;,.                  Swap: 0 B / 8.00 GiB (0%)Disk (/): 1.11 TiB / 1.82 TiB (61%) - btrfsDisk (/mnt/data_1): 534.41 GiB / 931.06 GiB (57%) - xfsDisk (/mnt/data_2): 525.78 GiB / 931.06 GiB (56%) - xfsLocal IP (br0): 192.168.1.118/24Locale: zh_CN.UTF-8

二、安装准备

安装前置依赖:

sudo dnf install dnf-plugin-config-manager python3-setuptools python3-wheel

如果先前已安装过 Fedora ROCm 软件包,最好先卸载,因为与 AMD 的软件包相比,其存在不兼容的依赖关系,而且不会直接升级:

sudo dnf remove rocm rocm-*

将 ROCm 仓库添加 yum 配置中:

sudo tee /etc/yum.repos.d/rocm.repo <<EOF
[ROCm]
name=rocm
baseurl=https://repo.radeon.com/rocm/el9/latest/main/
enabled=1
priority=50
gpgcheck=1
gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key[AMDGraphics]
name=rocmgraphics
baseurl=https://repo.radeon.com/graphics/latest/el/9.6/main/x86_64/
enabled=1
priority=50
gpgcheck=1
gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key
EOF

三、安装 ROCm

使用下方命令安装 ROCm 及其依赖:

sudo dnf install rocm rocm-developer-tools hipblas-devel hip-devel rocwmma-devel rocm-opencl-devel --allowerasing

--allowerasing 参数表示允许用 ROCm 仓库中提供的新版本替换先前安装的 opencl 或较旧的 ROCm 包。

输入 rocm-smi 验证是否安装成功。

此外,安装以下软件包方便对 AMD GPU 进行监控:

sudo dnf install radeontop

对于 amd-smi 工具,需要手动安装 Python 库:

删除之前的 AMD SMI 安装:

python3 -m pip list | grep amd
python3 -m pip uninstall amdsmi

从已经安装的 ROCm 实例安装 AMD SMI Python 库:

cd /opt/rocm/share/amd_smi
sudo python3 -m pip install --user .

在某些系统上,如果 pip 安装失败,在后面加上 --break-system-packages 参数

验证:

╰─ python3
Python 3.14.3 (main, Mar 26 2026, 00:00:00) [GCC 15.2.1 20260123 (Red Hat 15.2.1-7)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import amdsmi
>>>

amd-smi 工具:

╰─ amd-smi
+------------------------------------------------------------------------------+
| AMD-SMI 26.2.2+671d39a71e    amdgpu version: Linuxver ROCm version: 7.2.2    |
| VBIOS version: 00077014                                                      |
| Platform: Linux Baremetal                                                    |
|-------------------------------------+----------------------------------------|
| BDF                        GPU-Name | Mem-Uti   Temp   UEC       Power-Usage |
| GPU  HIP-ID  OAM-ID  Partition-Mode | GFX-Uti    Fan               Mem-Usage |
|=====================================+========================================|
| 0000:09:00.0  AMD Radeon RX 7700 XT | 1 %      35 °C   0             8/200 W |
|   0       0     N/A             N/A | 21 %     0.0 %            489/12272 MB |
+-------------------------------------+----------------------------------------+
+------------------------------------------------------------------------------+
| Processes:                                                                   |
|  GPU        PID  Process Name          GTT_MEM  VRAM_MEM  MEM_USAGE     CU % |
|==============================================================================|
|  No running processes found                                                  |
+------------------------------------------------------------------------------+

若需要在 ROCm 上使用 FlashAttention, 请设置以下环境变量:

export TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1

这将启用 AMD 实验性的即时编译 triton 数学库。

可将该环境变量写入 ~/.bashrc 来持久化。

四、安装 PyTorch

PyTorch 依赖 Python 3.10 或更新版本,使用以下命令安装:

pip3 install torch torchvision --index-url https://download.pytorch.org/whl/rocm7.2

安装成功后验证安装:

python -c "import torch; print(f'ROCm version: {torch.version.hip}'); print(f'GPU available: {torch.cuda.is_available()}')"

得到输出:

ROCm version: 7.2.26015
GPU available: True

PyTorch 安装完毕。

http://www.jsqmd.com/news/679067/

相关文章:

  • Leather Dress Collection惊艳效果:Leather Beltbra MicroShorts微距质感表现
  • 如何在Windows上直接运行Android应用?APK Installer让你轻松告别模拟器
  • 如何让微信在手机和平板同时在线?WeChatPad技术方案深度解析
  • 如何简化多表关联SQL逻辑_利用视图封装复杂JOIN操作
  • 如何高效使用文泉驿微米黑:开源中文字体的完整实践指南
  • Pixel Aurora Engine实际项目:为Retro Game Jam快速生成200+原创素材
  • 消除二叉树中的节点(python)
  • Dism++终极指南:告别系统维护焦虑的专业解决方案
  • 从嵌入式到终端:C语言万年历的5种实用改造方案
  • FUXA工业级可视化监控系统:5天从零构建专业SCADA平台的完整指南
  • Dify微调全流程详解:从数据准备、LoRA配置到评估上线,手把手带跑通首个可商用模型
  • 终极指南:3分钟掌握Applite,让Mac软件管理变得简单高效
  • 专知智库高企管理成熟度自诊系统:为政府/园区打造的高新技术企业培育管理利器
  • 如何让AMD/Intel显卡免费获得CUDA加速:ZLUDA兼容层完整指南
  • 大模型Agent的5种核心设计模式深度解析:工程师必备实战指南!从ReAct到Multi-Agent,手把手拆解!
  • Kotlin类型魔法:Any、Unit、Nothing 深度探秘
  • 2026Q2工业平移门技术全解析:甘肃工业门/兰州人行通道闸/兰州伸缩门/兰州保温卷帘门/兰州卷帘门/兰州工业厂房门/选择指南 - 优质品牌商家
  • 模型加载慢、推理卡顿、GC抖动频发,.NET 11 AI应用性能瓶颈诊断与7步精准修复,90%开发者忽略的第4步
  • 如何用NoFences告别桌面混乱:一个普通用户的桌面整理革命
  • HTML函数在多显示器环境下效率高吗_多屏硬件配置建议【汇总】
  • 告别臃肿!Elasticsearch平替Manticore登场
  • 掌握Agent智能体项目管理秘籍:让AI成为你的得力助手!
  • MySQL跨版本迁移数据格式不兼容_使用mysqldump全量导出导入
  • TranslucentTB开机自启动失效:Windows启动机制深度解析与系统级解决方案
  • 深度解析HsMod:基于BepInEx的炉石传说高级插件开发实战指南
  • 【flutter for open harmony】Flutter 聊天应用实战:go_router 路由管理完全实现指南
  • 宝塔面板网站备份文件太大怎么办_调整备份压缩等级与频率
  • CLIP-GmP-ViT-L-14环境部署:Ubuntu 22.04 + Python3.10 + torch2.3一键适配
  • 5步深度解析:ComfyUI-SUPIR图像超分辨率实战指南
  • 压缩软件应该选RAR格式还是ZIP格式?高压缩率高安全VS高兼容性之争的何去何从?