AMD显卡也能玩CUDA?这位开发者真是个“转接头”大侠!
  • 7908

你有没有想过,有一天,AMD显卡也能像NVIDIA的显卡那样,玩弄CUDA这位大侠?没错,最近这位名叫Speedstu的独立开发者,干了一件非常了不起的事情:他把“CUDA-for-AMD-Windows”这个神奇的魔法咒语给开源了!简直就像是给AMD的显卡装上了一个“万能转接头”。

想象一下,过去AMD显卡就像是不能参加派对的墙角羊,而现在,它们突然发现了通向VIP区域的隐秘通道,再也不需要为了追逐CUDA,忍受双系统的折磨。只要这条新渠道顺利通行,AMD的RX 9060 XT英雄显卡就能在Windows上为220万参数的PPO强化学习网络开荒,灯光下英姿飒爽,简直不亚于一场科技秀!

这个项目其实就像是一系列巧妙的PowerShell脚本,把ZLUDA的翻译层和AMD的HIP/ROCm SDK完美配对。就像调皮的小孩用拼图把玩具组装在一起,这些脚本自动搜索GPU架构,然后从冷冰冰的ZLUDA库中找来特定版本,轻轻一“映射”,就能与Windows上的ROCm数学库亲密接触。 星空体育

实测数据也很有趣,RX 9060 XT成功在官方的ZLUDA与AMD HIP SDK 6.4的搭配下,高喊出“13278 steps/秒”的威风;而从旧版ZLUDA二进制文件“拯救”出来的自定义层,表现稍逊,13276 steps/秒。不过,要是开发者的后续工作能够把LibTorch和ZLUDA的束缚给去掉,你会看到更疯狂的吞吐量,从此,RX 9060 XT犹如飞奔的奔驰车。

about image

当然,别高兴得太早,CUDA的护城河可不是随便可以填平的。可怜的cuDNN、TensorRT和NCCL这些关键库依然困在牢笼之中,如果某些高需求的工具过于依赖这些库,那它们就会像是“忘了钥匙被锁在家里”的小狗——无能为力。

但这一切揭示了一个至关重要的秘密:在AMD显卡上运行CUDA软件的障碍,并不是来自硬件本身,而是那一层可以被攻破的语言障碍。而且这个项目全开源,社区的小伙伴们未来很可能会集结,扩展兼容更多硬件,真是值得期待喝彩! 星空体育

所以,AMD用户们,快把你们的显卡拿出来,准备“翻译”一番,未来的科技舞台上,绝对少不了你们的身影!