AMD上个月已经放出了Ubuntu Linux的ROCm 5.7.1驱动更新,为Radeon Pro W7900和Radeon RX 7900 XTX显卡带来了ROCm和PyTorch支持。而在12月初的AMD Advancing AI活动召开之前,AMD宣布ROCm和PyTorch支持已经开放给同是RDNA 3架构的RX 7900 XT显卡,涵盖了RDNA 3架构的所有顶级产品。
AMD软件产品经理Erik Hultgren表示:我们对我们产品组合中的最新产品感到兴奋,这些高端GPU与ROCm结合,从软件和硬件的角度使人工智能变得更容易使用,因此开发人员可以选择最适合其需求的解决方案。RX 7900 XT拥有20GB的超大显存,Navi 31芯片上拥有总共168个AI加速单元,可提供快速的机械学习训练和推理性能。GPU还支持采用PyTorch的机械学习模型和演算法,这些模型和演算法转码针对AMD进行了最佳化,可以与运行22.04.3版本的Ubuntu Linux系统兼容。
不过,目前AMD还有大量其他的RDNA 3架构GPU并不支持相关应用,很多消费者希望能获得ROCm支持。相较而言NVIDIA一直在消费级显卡上提供较好的通用计算支持,CUDA也成为了不少消费者购买显卡考虑的因素,与之相比AMD的ROCm现在的支持相当有限,目前仍在持续优化当中。