开发者让CUDA在AMD显卡原生运行

IT时代网9月14日消息,据海外媒体报道,一名独立开发者在 Windows 环境下借助 ZLUDA 与 AMD 原生 HIP/ROCm SDK 的组合,让 CUDA 程序无需虚拟化或双系统启动即可直接在 AMD 显卡上运行。

这套高度自动化的 PowerShell 方案会自动识别用户的 GPU 架构、抓取特定版本的 ZLUDA,并将其映射到 Windows 中已有的 ROCm 数学库,成功把 CUDA 驱动接口以及 cuBLAS、cuSPARSE、cuFFT 等库直接对应到 AMD 等价实现。作为概念验证,作者在 AMD Radeon RX 9060 XT 上用未经修改的 CUDA 库端到端训练了一个 220 万参数的 PPO 强化学习网络。

对于想在 Windows 日常主机上原生试用那些硬编码依赖 CUDA 的实验性项目或专业 AI 工具、又不愿折腾 WSL2 透传的开发者来说,这条翻译管线提供了相当实用的路径。文档中的基准测试显示,在 Radeon RX 9060 XT 上运行同规模强化学习负载时,纯官方 ZLUDA 加 AMD 标准 HIP SDK 6.4 的中位吞吐约为每秒 13278 步。

编辑:林一舟

THE END