TensorFlow-DirectML

	Предупреждения
	h5py 3.0.0 и 3.1.0 не совместимы с TensorFlow. Перед использованием TensorFlow-DirectML убедитесь, что в вашей среде установлена другая версия h5py.

TensorFlow — это комплексная открытая платформа для машинного обучения. Данный репозиторий является ответвлением tensorflow, которое использует DirectML для обеспечения кроссплатформенного аппаратного ускорения на Windows 10 и подсистеме Windows для Linux (WSL). TensorFlow с DirectML позволяет обучать и делать выводы на основе сложных моделей машинного обучения на широком спектре совместимого с DirectX 12 оборудования.

Последний релиз:

Вопросы, проблемы и обратная связь

• Задайте вопрос: Обсуждения
• Сообщите об ошибке: Проблемы
• Узнайте о нашей дорожной карте: Вики

Вы также можете связаться с нами напрямую по адресу askdirectml@microsoft.com.

Начало работы

TensorFlow с DirectML поддерживается как последними версиями Windows 10, так и подсистемой Windows для Linux. Подробные инструкции по началу работы см. в документации по обучению ML с ускорением на GPU (docs.microsoft.com). Официальные пакеты Python доступны в проекте tensorflow-directml PyPI.

TensorFlow с DirectML всё ещё находится в активной разработке и находится в состоянии предварительного просмотра. Репозиторий DirectML включает несколько примеров, которые были протестированы для работы с последними сборками на PyPI. Эти примеры включают сценарии как для вывода, так и для обучения, и вы можете либо обучить модели с нуля, либо использовать предоставленные предварительно обученные веса. Однако мы рекомендуем протестировать любые модели, совместимые с TensorFlow 1.15 — если у вас возникнут проблемы, сообщите нам!

Следующие ресурсы предоставляют дополнительную информацию о DirectML и TensorFlow:

• DirectML GitHub
• RFC: TensorFlow на DirectML
• Домашняя страница TensorFlow

Системные требования

Windows 10

• Windows 10 версии 1709, 64-разрядная (сборка 16299 или выше)
• Python x86-64 3.5, 3.6 или 3.7¹
• Один из следующих поддерживаемых графических процессоров:
  • AMD Radeon R5/R7/R9 серии 2xx или новее
  • Intel HD Graphics 5xx или новее
  • NVIDIA GeForce GTX 9xx серии GPU или новее

¹ Примечание: Python 3.8 или более поздние версии в настоящее время не поддерживаются. Для использования официальных пакетов PyPi требуется интерпретатор CPython. NumPy 1.19.4 требует KB4598291 для правильной работы в Windows.

Подсистема Windows для Linux

• Windows 10 Insider Preview, 64-бит (Build 20150 или выше)
• Python x86-64 3.5, 3.6, или 3.7²
• Один из следующих поддерживаемых графических процессоров:
  • AMD Radeon R5/R7/R9 2xx series или новее, и драйвер WSL 20.20.01.05
  • Intel HD Graphics 6xx или новее, и драйвер WSL 28.20.100.8322
  • NVIDIA GeForce GTX 9xx series GPU или новее, и драйвер WSL 460.20

² Примечание: Python 3.8 или новее в настоящее время не поддерживается. Для использования официальных пакетов PyPi требуется интерпретатор CPython.