Hugging Face TRL 正式支持 AMD GPU,MI300X 跑全套测试
QGallouedec · x · 2026-10-21
Hugging Face 的 TRL 训练库正式宣布官方支持 AMD Instinct GPU。此前 AMD GPU「基本能用」，新变化是团队明确承诺支持并深度测试：每个 PR 都会在 MI300X 上运行完整测试套件。
TRL 官方文档说明：PyTorch 的 ROCm 构建会把 AMD GPU 暴露为 cuda 设备，因此无需改代码即可运行 trainer。安装时用 ROCm 版 PyTorch 加 TRL 即可；注意力内核方面，文档示例用的 FlashAttention 内核暂无 ROCm 构建，AMD GPU 上需改用 kernels-community/aiter-flash-attn(从 Hub 下载)。
「Infra」频道最新
- 英伟达 GTC 柏林站 keynote 开幕在即，排队队伍绵延数个街区 — TheZachMueller · 2026-10-21
- 曝 NVIDIA 欧洲大动作，现场屏幕展示生态版图 — testingcatalog · 2026-10-21
- 黄仁勋现身柏林 GTC 主题演讲，NVIDIA 官方直播 — nvidia · 2026-10-21
- PyTorch 基金会一周新增 9 名会员，Anyscale 升至白金级 — YvesMulkers · 2026-10-21
- 300 美元攒机跑本地 27B 模型：双 3060 12GB 方案求建议 — Loose_Doubt367 · 2026-10-21
- CoreWeave 背债 420 亿美元,占七大 Neocloud 债务近半(转推版) — FinanceYF5 · 2026-10-21