設置
  • 日夜間
    隨系統(tǒng)
    淺色
    深色
  • 主題色

微軟 TensorFlow-DirectML 正式版發(fā)布:可在 WSL 上 GPU 加速機器學習

2021/9/12 9:32:17 來源:IT之家 作者:姜戈 責編:姜戈

IT之家 9 月 12 日消息 微軟近日發(fā)布了 TensorFlow-DirectML 的正式版,該工具可讓用戶在 Windows 的 Linux 子系統(tǒng)(WSL)上 GPU 加速機器學習(ML)模型訓練。

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

IT之家了解到,用戶可以在 Python 環(huán)境下(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 驅(qū)動;

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

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

具體使用方法,大家可以查看 TensorFlow-DirectML 的 GitHub 頁面。

廣告聲明:文內(nèi)含有的對外跳轉(zhuǎn)鏈接(包括不限于超鏈接、二維碼、口令等形式),用于傳遞更多信息,節(jié)省甄選時間,結(jié)果僅供參考,IT之家所有文章均包含本聲明。

相關文章

關鍵詞:WSL,TensorFlow,DirectML,機器學習

軟媒旗下網(wǎng)站: IT之家 最會買 - 返利返現(xiàn)優(yōu)惠券 iPhone之家 Win7之家 Win10之家 Win11之家

軟媒旗下軟件: 軟媒手機APP應用 魔方 最會買 要知