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

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

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

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

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