Folia
← 返回头版

Windows平台上AMD GPU实现CUDA兼容运行

一个开源项目通过整合ZLUDA和AMD ROCm/HIP技术栈,实现了在Windows系统中于AMD显卡上直接运行CUDA目标应用的兼容性方案1。该方案已在AMD Radeon RX 9060 XT上进行了验证1,支持cuBLAS、cuSPARSE、cuFFT等多个CUDA库的调用1,并通过包含2,216,347个参数的PPO神经网络完成前向推理和学习任务的测试1。

该实现采用了ZLUDA v6-preview.69、AMD HIP SDK 6.4和LibTorch 2.3.0 + cu118等核心版本组件1。性能对比数据显示,上游路径达到13,278 median SPS,而自定义覆盖方案的性能为12,876 SPS1。不过该方案仍存在明显局限,目前仅在RX 9060 XT上经过验证1,且不支持cuDNN/MIOpen、NCCL、TensorRT以及某些自定义CUDA扩展1。


来源

  1. Hacker NewsCUDA for AMD on Windows

评论