Включение PyTorch с помощью DirectML в WSL 2

PyTorch с DirectML предоставляет простой способ использования для разработчиков, чтобы попробовать последние и величайшие модели искусственного интеллекта на своем Windows компьютере. Вы можете скачать PyTorch с помощью DirectML, установив пакет PyPi torch-directml. После настройки вы можете начать с наших примеров или использовать набор средств Foundry для VS Code.

Проверьте версию Windows

Пакет torch-directml в подсистема Windows для Linux (WSL) 2 работает начиная с Windows 11 (сборка 22000 или выше). Номер версии сборки можно проверить, выполнив команду winver с помощью команды Run (клавиша логотипа Windows + R).

Проверка обновлений драйверов GPU

Убедитесь, что установлен последний драйвер GPU. Выберите Check for updates в разделе клиентский компонент Центра обновления Windows Windows Settings.

Настройка Torch-DirectML

Установка WSL 2

Чтобы установить подсистема Windows для Linux (WSL) 2, см. инструкции в Install WSL.

Затем установите драйвер графического интерфейса WSL, следуя инструкциям в файле README.md в репозитории microsoft/wslg GitHub.

Настройка среды Python

Рекомендуется настроить виртуальную среду Python в WSL 2. Существует множество средств, которые можно использовать для настройки виртуальной среды Python. В этом разделе мы будем использовать Miniconda. В остальной части этой установки предполагается, что вы используете среду Miniconda.

Установите Miniconda, выполнив инструкции по установщику Linux на сайте Anaconda или выполнив следующие команды в WSL 2.

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

После установки Miniconda создайте среду Python с именем pytdml и активируйте ее с помощью следующих команд:

conda create --name pytdml -y
conda activate pytdml

Установка PyTorch и Torch-DirectML

Все, что необходимо для установки, — установить последнюю версию torch-directml, выполнив следующую команду:

pip install torch-directml

Проверка и создание устройства

После установки пакета torch-directml можно убедиться, что он работает правильно, добавив два тензора. Сначала запустите интерактивный сеанс Python и импортируйте Факел со следующими строками:

import torch
import torch_directml
dml = torch_directml.device()

Текущий выпуск torch-directml сопоставляется с бэкендом "PrivateUse1". API torch_directml.device() — это удобная оболочка для отправки тензоров на устройство 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

PyTorch с примерами и отзывами о DirectML

Ознакомьтесь с нашими примерами , чтобы узнать больше об использовании PyTorch с DirectML. Если у вас возникли проблемы или есть отзывы о пакете PyTorch с помощью DirectML, обратитесь к нашей команде.