近期，微軟發布了一款名為TensorFlow-DirectML的TensorFlow擴展套件，幫助開發者能夠在Windows平臺上通過支持DirectX?12的GPU來加速TensorFlow模型運算，并同樣適用Linux的Windows子系統（WSL）。目前該套件已經在GitHub開源，供開發者取用程序碼。

此前微軟便在build?2020大會上曾宣布，適用于Linux的Windows子系統WSL?2（Windows?Subsystem?for?Linux?2）開始支持GPU硬件加速。但微軟表示，TensorFlow雖可以同時利用CPU和GPU來執行運算，但GPU加速僅限于特定供應商的平臺。

另一方面DirectX一直以來，被廣泛地用于加速Windows上的游戲和計算應用程序，其DirectML可以讓支援DirectX?12的GPU，執行高效能的數學運算。因此微軟透過整合TensorFlow以及DirectML，讓開發者可以在Windows，使用GPU執行完整的機器學習訓練和推理功能。

TensorFlow-DirectML是通過加入名為DML而非GPU的新裝置，將TensorFlow和DirectML整合在一起，其新的核心集建構在DirectML?API之上，而非象是現有的CPU和GPU核心，建立在Eigen函式庫之上。

由于DirectML又是以Direct3D?12低階函式庫開發，該繪圖API是專為需要對資源分配和工作排程，進行絕對控制的高效能低延遲應用程序設計，因此DirectML與TensorFlow整合，還需要仰賴裝置Runtime負責在裝置與主機間，管理裝置存儲器、復制張量、紀錄GPU命令，以及排程工作。
責任編輯:pj