1. 程式人生 > 資訊 >微軟 TensorFlow-DirectML 正式版釋出:可在 WSL 上 GPU 加速機器學習

微軟 TensorFlow-DirectML 正式版釋出:可在 WSL 上 GPU 加速機器學習

9 月 12 日訊息微軟近日釋出了TensorFlow-DirectML 的正式版,該工具可讓使用者在 Windows 的 Linux 子系統(WSL)上 GPU 加速機器學習(ML)模型訓練。

顧名思義,TensorFlow-DirectML 用於將谷歌TensorFlow 機器學習平臺與微軟DirectML 機器學習 API 適配,以支援在 Win10/WSL 上進行機器學習模型訓練時獲得 GPU 加速

瞭解到,使用者可以在 Python 環境下(Python 3.8 及以上暫不支援)輸入 pip install tensorflow-directml 進行安裝,支援在各種 DirectX 12 相容硬體上訓練和推理複雜的機器學習模型。

Win10 軟硬體要求:

  • Windows 10 1709,64 位(Build 16299 或更高);

  • Python x86-64 3.5、3.6 或 3.7;

以下支援的 GPU 之一:

  • AMD Radeon R5/R7/R9 2xx 系列或更新版本;

  • 英特爾 HD Graphics 5xx 或更新版本;

  • 英偉達 GeForce GTX 9xx 系列 GPU 或更新版本;

WSL 軟硬體要求:

  • Windows 10 Insider Preview,64 位(Build 20150 或更高);

  • Python x86-64 3.5、3.6 或 3.7;

以下支援的 GPU 之一:

  • AMD Radeon R5/R7/R9 2xx 系列或更新版本,以及 20.20.01.05 WSL 驅動;

  • 英特爾 HD Graphics 6xx 或更新版本,以及 28.20.100.8322 WSL 驅動;

  • 英偉達 GeForce GTX 9xx 系列或更新版本,以及 460.20 WSL 驅動;

具體使用方法,大家可以檢視 TensorFlow-DirectML 的 GitHub 頁面