在 AMD 显卡上运行 CUDA 程序
CUDA for AMD on Windows
一个名为 CUDA-for-AMD-Windows 的开源项目,利用 ZLUDA 和 AMD HIP/ROCm 技术,成功让原本专为 NVIDIA 设计的 CUDA 应用程序在 AMD GPU 上运行。该项目已在 AMD Radeon RX 9060 XT 上验证,支持 LibTorch 等计算负载,并完成了包含 220 万参数的 PPO 网络训练。虽然目前仅验证了特定硬件,且 cuDNN 等部分库尚未完全支持,但这为 Windows 用户提供了在 AMD 显卡上运行 CUDA 生态的可行方案,打破了硬件厂商的生态壁垒。
一个真实的包含 2,216,347 个参数的 PPO 网络在面向 CUDA 的设备上完成了前向推理、PPO 学习和优化器工作。