微軟攜手AMD推動基於RDNA 2 GPU的TensorFlow-DirectML訓練
作為一款機器學習(ML)工具,TensorFlow-DirectML 被設計適用於廣泛的硬體類型(尤其是 GPU)。 最新消息是,AMD 已同微軟攜手,為 Windows 平台帶來了基於 RNDA 2 的新體驗。 具體說來是,其推理性能提升到了 3.7 倍、整體 AI Benchmark Alpha 跑分也達到了 4.4 倍。
(來自:PYPI.org官網)
據悉,這是在AMD Radeon RX 6900 XT / RX 6600 XT 圖形硬體上展開的測試,且訓練成績最高提升到了 4.4 倍。
微軟指出,為應對世界上最嚴苛的挑戰,使用者需要這些功能強大、易於訪問、且適用於各種硬體類型的機器學習工具。
比如科學家和研究人員們可以利用數據中心的應用程式,而學生與專業人士亦可利用其已擁有的硬體(筆記本 / 桌上型機)來開發訓練 ML 模型。
基於客戶的反饋,大家想要找到能夠更好地利用現代Windows PC 硬體來運行機器學習工作流的解決方案。
所以為了讓 Windows 用戶能夠更輕鬆地使用 ML 工具,微軟在去年宣佈了支援 GPU 加速訓練工作流的預覽版本,輔以可在 Windows Linux 子系統(WSL)中使用的 DirectML 機器學習支援框架。
這種 GPU 加速訓練適用於任何交融 DirectX 12 的 GPU,且完全支援 AMD 旗下的 Radeon / Radeon PRO 顯卡產品線。
微軟指出,這進一步壯大了在基於 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項目介紹