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

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

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

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

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

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

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

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

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

Скачайте и установите на систему установщик Miniconda для Windows. Дополнительные рекомендации по настройке на сайте Anaconda. После установки 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, обратитесь к нашей команде.