微軟攜手AMD推動基於RDNA 2 GPU的TensorFlow-DirectML訓練

作為一款機器學習(ML)工具,TensorFlow-DirectML 被設計適用於廣泛的硬件類型(尤其是 GPU)。最新消息是,AMD 已同微軟攜手,為 Windows 平台帶來了基於 RNDA 2 的新體驗。具體說來是,其推理性能提升到了 3.7 倍、整體 AI Benchmark Alpha 跑分也達到了 4.4 倍。

0.png

(來自:PYPI.org 官網)

據悉,這是在 AMD Radeon RX 6900 XT / RX 6600 XT 圖形硬件上展開的測試,且訓練成績最高提升到了 4.4 倍。

微軟指出,為應對世界上最嚴苛的挑戰,用戶需要這些功能強大、易於訪問、且適用於各種硬件類型的機器學習工具。

1.jpg

比如科學家和研究人員們可以利用數據中心的應用程序,而學生與專業人士亦可利用其已擁有的硬件(筆記本 / 台式機)來開發訓練 ML 模型。

基於客戶的反饋,大家想要找到能夠更好地利用現代 Windows PC 硬件來運行機器學習工作流的解決方案。

2.jpg

所以為了讓 Windows 用戶能夠更輕鬆地使用 ML 工具,微軟在去年宣布了支持 GPU 加速訓練工作流的預覽版本,輔以可在 Windows Linux 子系統(WSL)中使用的 DirectML 機器學習支持框架。

這種 GPU 加速訓練適用於任何交融 DirectX 12 的 GPU,且完全支持 AMD 旗下的 Radeon / Radeon PRO 顯卡產品線。

3.gif

微軟指出,這進一步壯大了在基於 AMD Radeon 顯卡 + Microsoft Windows 10 操作系統的設備平台上的機器學習模型開發能力。

這些更新將使您受益於微軟和 AMD 在整個 TensorFlow-DirectML 開發過程中實現的所有性能改進,並將在使用 AMD 顯卡時,讓您的 Windows 機器學習工作流程提升到一個全新的水平。

最後,如果你手頭已有兼容 DirectX 12 的 AMD Radeon 圖形硬件(例如前面提到過的 RX 6900 XT / 6600 XT 顯卡)、並且安裝了最新版的 AMD Radeon Software Adrenalin 驅動程序,那現在就可立即開始使用 TensorFlow-DirectML 。

● AMD Radeon Software Adrenalin 驅動下載頁面

● Microsoft Windows AI Platform 項目介紹

(0)
上一篇 2021-09-16 11:24
下一篇 2021-09-16 11:24

相关推荐