摘要:
AMD近期宣布了面向Ubuntu Linux的ROCm 5.7.1驱动程序的发布,此版本特别为RDNA 3架构的Radeon Pro W7900及Radeon RX 7900 XTX显卡配备了PyTorch 2.0.1的支持。此举旨在推动研究人员充分利用AMD Instinct加速器的潜能,促进高性能计算和AI创新。
详细内容:
AMD已推出针对Ubuntu Linux操作系统的ROCm 5.7.1驱动程序,这一更新为基于RDNA 3架构的Radeon Pro W7900及Radeon RX 7900 XTX显卡带来了PyTorch 2.0.1的兼容性。此举不仅增强了开发者利用AMD消费级旗舰GPU进行AI应用开发的能力,也降低了对NVIDIA产品的依赖。
优化的ROCm 5.7.1驱动程序特别针对PyTorch 2.0.1进行了调整,以充分挖掘AMD RDNA3架构GPU的功能特性。然而,目前仅支持Radeon RX 7900 XTX 24GB及Radeon Pro W7900 48GB这两款顶级显卡,主要考虑到大型语言模型(LLM)等AI应用通常需要大量显存资源的需求,同时,AMD希望通过推广高端硬件来增加销量。
值得一提的是,Navi 31 GPU内置了192个AI加速器,相较于Navi 21的96个实现了翻倍,使其成为适合AI应用的理想选择。AMD Radeon RX 7900 XTX 24GB的定价为999美元(图灵汇注:国行首发价为7999元,当前市价普遍低于官方指导价),相对于专业AI和HPC GPU而言更为经济实惠。
这款显卡配置包括96CU、6144个流处理器,GPU频率范围在1900-2500MHz之间,单精度性能达到61TFLOPS。其显存配置为24GB 20Gbps GDDR6,带宽高达960GB/s,搭配96MB无限缓存,等效带宽可达3.5TB/s。