在 WSL 2 上使用 DirectML 啟用 PyTorch

發行項
05/21/2024

PyTorch 與 DirectML 提供易於使用的方式，讓開發人員在 Windows 計算機上試用最新且最偉大的 AI 模型。您可以安裝 torch-directml PyPi 套件，以使用 DirectML 下載 PyTorch。設定之後，您就可以開始使用我們的範例，或使用適用於 VS Code 的 AI 工具組。

檢查您的 Windows 版本

Windows 子系統 Linux 版（WSL） 2 中的 torch-directml 套件從 Windows 11 開始運作（組建 22000 或更高版本）。您可以透過 winver [執行 ] 命令執行來檢查組建版本號碼（Windows 標誌鍵 + R）。

檢查 GPU 驅動程式更新

請確定您已安裝最新的 GPU 驅動程式。在設定應用程式的 [Windows Update] 區段中，選取 [檢查更新]。

設定 Torch-DirectML

安裝 WSL 2

若要安裝 Windows 子系統 Linux 版（WSL） 2，請參閱安裝 WSL 中的指示。

然後遵循 microsoft/wslg GitHub 存放庫中檔案中的README.md指示來安裝 WSL GUI 驅動程式。

設定 Python 環境

建議您在 WSL 2 內設定虛擬 Python 環境。有許多工具可用來設定虛擬 Python 環境，在此主題中，我們將使用 Anaconda 的 Miniconda。此設定的其餘部分假設您使用Miniconda環境。

遵循 Anaconda 網站上的 Linux 安裝程式指引，或在 WSL 2 中執行下列命令，以安裝 Miniconda。

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh 
bash Miniconda3-latest-Linux-x86_64.sh

安裝 Miniconda 之後，請建立名為 pytdml 的 Python 環境，並透過下列命令加以啟用：

conda create --name pytdml -y
conda activate pytdml

安裝 PyTorch 和 Torch-DirectML

注意

torch-directml 套件最多可支援 PyTorch 2.2。

取得安裝程式所需的所有專案，都是執行下列命令來安裝最新版本的 torch-directml ：

pip install torch-directml

驗證和裝置建立

安裝 torch-directml 套件之後，您可以藉由新增兩個張張量來確認它是否正常執行。首先啟動互動式 Python 工作階段，並使用下列幾行匯入 Torch：

import torch
import torch_directml
dml = torch_directml.device()

目前版本的 torch-directml 會對應至 “PrivateUse1” Torch 後端。 torch_directml.device（） API 是將張量傳送至 DirectML 裝置的便利包裝函式。

建立 DirectML 裝置後，您現在可以定義兩個簡單的張量;一個包含 1 的張量，另一個則包含 2。將張量放在「dml」裝置上。

tensor1 = torch.tensor([1]).to(dml) # Note that dml is a variable, not a string!
tensor2 = torch.tensor([2]).to(dml)

將張量加在一起，並列印結果。

dml_algebra = tensor1 + tensor2
dml_algebra.item()

您應該會看到輸出的數位 3，如下列範例所示。

>>> import torch
>>> tensor1 = torch.tensor([1]).to(dml)
>>> tensor2 = torch.tensor([2]).to(dml)
>>> dml_algebra = tensor1 + tensor2
>>> dml_algebra.item()
3

使用 DirectML 範例和意見反應的 PyTorch

請參閱我們的範例，以查看更多使用 PyTorch 搭配 DirectML。如果您遇到問題，或有使用 DirectML 套件的 PyTorch 意見反應，請在這裡與我們小組聯繫。

共用方式為