在最新版的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。

关键词: AMD 微软 RDNA2