TensorFlow-DirectML-Plugin is an open-source DirectML backend for TensorFlow.

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for patvigno from gravatar.com patvigno Avatar for wumaggie from gravatar.com wumaggie

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: MIT License (MIT)
  • Author: Microsoft
  • Tags tensorflow , tensor , machine , learning , plugin
Classifiers
Report project as malware

Project description

Accelerate machine learning training with TensorFlow on Windows and Windows Subsystem for Linux (WSL).

TensorFlow-DirectML-Plugin leverages TensorFlow’s Pluggable Device API to create a DirectML backend for hardware-accelerated machine learning workflows on any DirectX 12-compatible GPU.

Install the base tensorflow and tensorflow-directml-plugin packages to get started. Check out the GitHub repository to build from source.

TensorFlow, the TensorFlow logo, and any related marks are trademarks of Google Inc.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page