共用方式為

使用 tensorflow-directml-plugin 啟用 TensorFlow 2 的 GPU 加速

這很重要

此專案現已停止，並未積極開發。

此版本提供學生、初學者和專業人員使用 TensorFlow 2 的 DirectML 外掛程式，在現有已啟用 DirectX 12 的硬體上執行機器學習 （ML） 訓練的方式。

備註

您可以使用 Python x86-64 3.10 進行安裝 tensorflow-directml-plugin 。 但 tensorflow-directml-plugin 不支援 3.11 版及更新的版本。

瞭解如何使用 tensorflow-directml-plugin 來設定您的裝置，以配合 GPU 執行和訓練模型。

步驟 1：最低（和最大值）系統需求

安裝 TensorFlow-DirectML-Plugin 之前，請確定您的 Windows 或 WSL 版本支援 TensorFlow-DirectML-Plugin。

Windows 原生

  • Windows 10 版本 1709、64 位（組建 16299 或更高版本）或 Windows 11 版本 21H2、64 位 （組建 22000 或更高版本）
  • Python x86-64 3.7、3.8、3.9 或 3.10。 3.10 版也是最大支援的版本。
  • 支援的 GPU 中之一：
    • AMD Radeon R5/R7/R9 2xx 系列或更新
    • Intel HD Graphics 5xx 或更新版
    • NVIDIA GeForce GTX 9xx 系列 GPU 或更新版本

Windows 子系統 Linux 版

安裝最新的 GPU 驅動程式

請確定您已為硬體安裝最新的 GPU 驅動程式。 選取 [檢查更新] 在 Windows 設定的 [Windows 更新] 區段中。 如有需要，請使用上述連結，從硬體廠商取得安裝。

步驟 2：設定 Windows 環境

Windows 原生

原生 Windows 上的 TensorFlow-DirectML-Plugin 套件從 Windows 10 版本 1709 開始運作（組建 16299 或更高版本）。 您可以透過 winver[執行 ] 命令執行來檢查組建版本號碼（Windows 標誌鍵 + R）。

Windows 子系統 Linux 版

安裝上述驅動程序之後，請確定您 啟用 WSL 並 安裝以 glibc 為基礎的發行版 （例如 Ubuntu 或 Debian）。 針對我們的測試，我們使用Ubuntu。 在 Windows 設定的「Windows 更新」部分中選擇「檢查更新」，確保您擁有最新的核心。

備註

請確定您已啟用 更新 Windows 時接收其他 Microsoft 產品的更新 選項。 您可以在 Windows 設定的「Windows 更新」部分的「進階選項」中找到它。

針對這些功能，您需要 5.10.43.3 或更高版本的核心版本。 您可以在 PowerShell 中執行下列命令來檢查版本號碼。

wsl cat /proc/version

步驟 3：設定您的環境

建議您在 Windows 內設定虛擬 Python 環境。 有許多工具可用來設定虛擬 Python 環境，如需這些指示，我們將使用 Anaconda 的 Miniconda。 此設定的其餘部分假設您使用Miniconda環境。 深入瞭解如何使用 Python 環境

在 Miniconda 中建立環境

在您的系統上下載並安裝 Miniconda Windows 安裝程式 。 Anaconda 網站上有 其他設定指引 。 安裝 Miniconda 之後，請使用名為 tfdml_plugin 的 Python 建立環境，然後透過下列命令加以啟用。

conda create --name tfdml_plugin python=3.9 

conda activate tfdml_plugin

備註

tensorflow 版本 >= 2.9，python 版本 >= 3.7 支援。

步驟 4：安裝基底 TensorFlow

下載基底 TensorFlow 套件。 目前 directml 外掛程式僅適用於 tensorflow–cpu==2.10 和 而非 tensorflow 或 tensorflow-gpu。

pip install tensorflow-cpu==2.10

步驟 5：安裝 tensorflow-directml-plugin

安裝此套件會自動啟用現有腳本的 DirectML 後端，而不需要變更任何程式代碼。

pip install tensorflow-directml-plugin

備註

如果您的定型腳本將裝置字串硬編碼為非 "GPU"，可能會產生錯誤。

或者，您可以從來源建置套件。 從來源建置 tensorflow-directml-plugin 的指示。

TensorFlow 搭配 DirectML 的範例與回饋意見

請查看 我們的範例 ，或使用您現有的模型腳本。 如果您遇到問題，或有 TensorFlow-DirectML-Plugin 套件的意見反應，請 與我們小組連線。

  • Last updated on