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

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

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

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

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