HeadlinesBriefing favicon HeadlinesBriefing.com

通过 ZLUDA + ROCm/HIP 在 AMD GPU 上运行 CUDA 应用

Hacker News •
×

一个可工作的可复现堆栈现在可通过 ZLUDA + ROCm/HIP,在 AMD GPU 上运行面向 CUDA 的 Windows 应用程序。该设置支持启用 CUDA 的 Lib Torch 工作负载,并已在 AMD Radeon RX 9060 XT (gfx1200) 上完成验证。关键组件包括 ZLUDA v6-preview.69、AMD HIP SDK 6.4,以及 Lib Torch 2.3.0 + cu118。包括 cuBLAS、cuBLASLt、cuSPARSE 和 cuFFT 在内的所有 CUDA driver APIs 均通过 cuda_check。一个 2.2M-parameter PPO network 在 CUDA-facing device 上成功完成了 forward/inference、PPO learning 和 optimizer operations。使用此 repository 生成的 runtime 完成了一次干净的 validation iteration,共 65,536 timesteps。

安装需要 AMD prerequisites:适用于 Windows 的当前 GPU driver 和 HIP SDK 6.4。用户 clone 此 repository 并运行 install.ps1;该脚本会检测 GPU、验证 dependencies、下载 ZLUDA 和 Lib Torch、验证 SHA-256 hashes,并运行 cuda_check.exe。runtime 可以被 staged,也可以通过 run-zluda.ps1 启动。Diagnostic scripts 包括 doctor.ps1、gpu-scan.ps1 和 test-runtime.ps1。

GPU compatibility 目前仅限于 RX 9060 XT;其他 AMD GPUs 是候选对象,但不保证可用。GPU scanner 会记录 model、gfx architecture、driver 和 HIP info,且不会收集 usernames 或 tokens。由于缺少 MIOpen,stable HIP SDK 无法使用 CUDA DNN,不过 dense/GEMM-heavy Lib Torch training 仍可能正常运行。