谁说只能玩游戏?联手微软AMD为A卡提升AI性能

近日,在最新版的AMD显卡驱动中,AMD联手微软大幅提升了RDNA2架构显卡的AI性能,其中RX 6900 XT、RX 6600 XT的AI推理性能可分别提升最多3.7倍和3.4倍。

不久前,微软将TensorFlow带到了Windows平台上,并发布了TensorFlow-DirectML。TensorFlow是谷歌推出的一个端到端开源机器学习平台,微软通过将其与微软DirectML机器学习API适配,让用户能够在Windows 10以及其自带的Linux子系统上使用GPU加速机器模型训练。

AMD的21.8.1版显卡驱动已经集成正式版的TensorFlow-DirectML,对比预览版的21.5.2驱动,RX 6900 XT、RX 6600 XT的AI推理性能可分别提升最多3.7倍、3.4倍,AI Benchmark Alpha测试得分则可分别提升4.4倍、3.7倍。

微软表示,GPU加速的TensorFlow-DirectML机器学习训练支持所有DX12兼容显卡,AMD RDNA2架构的Radeon RX、Radeon PRO系列显卡已经率先完全支持,操作系统则支持Windows10、Windows 11。

(7767665)

相关文章