2024-01-25 11:08:42

AMD希望通过在其RadeonRX7900XTX和RadeonProW7900显卡中添加对PyTorch的支持，让开发人员和研究人员更容易使用AI。

基于其RDNA3GPU架构，这些是目前最好的GPU，它们现在可以让用户为机器学习训练和推理建立私有且经济高效的工作流程。以前，用户可能需要依靠云访问兼容的GPU来处理此类AI工作负载。

Radeon产品管理副总裁DanWood表示：“我们很高兴能够使用基于AMDRadeonRX7900XTX和RadeonProW7900GPU以及ROCm开放软件平台构建的PyTorch为AI社区提供机器学习开发新支持。”“这是我们第一个基于RDNA3架构的实施，我们期待与社区合作。”

随着PyTorch机器学习框架现在在其最强大的显卡上得到支持，AMD希望为那些没有所需手段或基础设施的用户开放对AI工作负载的访问。

任何希望利用PyTorch的人都可以使用适用于GPU的Radeon开放计算(ROCm)软件堆栈，该软件堆栈涵盖通用计算、高性能计算(HPC)以及异构计算。

借助AMDROCm5.7，配备基于RDNA3GPU以及CDNAGPU和AMDInstinctMI系列加速器的机器的用户也可以使用PyTorch。

由于ROCm是开源的，开发人员可能希望采用各种不同的方向，并添加对自己特定AI处理需求的支持。例如，人们对在AMD加速处理单元(APU)上运行稳定扩散有着巨大的兴趣。