【智能日报】10月23日消息，AMD最新发布的ROCm 5.7.1驱动程序为Ubuntu Linux用户带来了一项重要更新，为基于RDNA 3架构的Radeon Pro W7900和Radeon RX 7900 XTX显卡增加了对PyTorch 2.0.1的全面支持。
这一举措对开发人员而言具有重大意义，因为他们现在可以充分利用AMD的消费级旗舰GPU，并运行基于PyTorch库的人工智能应用程序，无需过度依赖竞争对手NVIDIA的解决方案。
据智能日报了解，这次的ROCm 5.7.1驱动程序专门针对PyTorch 2.0.1进行了优化，确保其充分发挥AMD RDNA3架构GPU的潜力。
然而，当前ROCm 5.7.1驱动程序仅支持两款旗舰显卡，即Radeon RX 7900 XTX 24GB和Radeon Pro W7900 48GB。这主要是因为大型语言模型(LLM)等人工智能应用通常需要大容量显存，而AMD也希望能够促进高端硬件的销售。
AMD还指出，Navi 31 GPU内置了192个AI加速器，相对于Navi 21的96个加速器实现了翻倍，这使其非常适用于各种人工智能应用。与此同时，AMD Radeon RX 7900 XTX 24GB的定价令人惊喜，售价为999美元(国内首发价为7999元，目前市场价普遍低于该价格)，远低于专业AI和高性能计算GPU的价格。
AMD Radeon RX 7900 XTX拥有96CU，6144个流处理器，GPU频率在1900-2500MHz之间，单精度性能可达61TFLOPS。显存容量为24GB，带宽高达20Gbps的GDDR6，总带宽可达960GB/s，再加上96MB的无限缓存，等效带宽达到3.5TB/s。这使其成为了人工智能领域的一项强大选择，同时为开发者提供了更多的灵活性和性能。
