Skip to main content

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

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.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

tensorflow_directml_plugin-0.1.0.dev220928-cp310-cp310-win_amd64.whl (9.0 MB view details)

Uploaded CPython 3.10 Windows x86-64

tensorflow_directml_plugin-0.1.0.dev220928-cp310-cp310-manylinux2010_x86_64.whl (26.8 MB view details)

Uploaded CPython 3.10 manylinux: glibc 2.12+ x86-64

tensorflow_directml_plugin-0.1.0.dev220928-cp39-cp39-win_amd64.whl (9.0 MB view details)

Uploaded CPython 3.9 Windows x86-64

tensorflow_directml_plugin-0.1.0.dev220928-cp39-cp39-manylinux2010_x86_64.whl (26.8 MB view details)

Uploaded CPython 3.9 manylinux: glibc 2.12+ x86-64

tensorflow_directml_plugin-0.1.0.dev220928-cp38-cp38-win_amd64.whl (9.0 MB view details)

Uploaded CPython 3.8 Windows x86-64

tensorflow_directml_plugin-0.1.0.dev220928-cp38-cp38-manylinux2010_x86_64.whl (26.8 MB view details)

Uploaded CPython 3.8 manylinux: glibc 2.12+ x86-64

tensorflow_directml_plugin-0.1.0.dev220928-cp37-cp37m-win_amd64.whl (9.0 MB view details)

Uploaded CPython 3.7m Windows x86-64

tensorflow_directml_plugin-0.1.0.dev220928-cp37-cp37m-manylinux2010_x86_64.whl (26.8 MB view details)

Uploaded CPython 3.7m manylinux: glibc 2.12+ x86-64

File details

Details for the file tensorflow_directml_plugin-0.1.0.dev220928-cp310-cp310-win_amd64.whl.

File metadata

File hashes

Hashes for tensorflow_directml_plugin-0.1.0.dev220928-cp310-cp310-win_amd64.whl
Algorithm Hash digest
SHA256 828e71f5dc794c5f4bbc6ad07dd5eecd964f6835dee7f9a5f009e29c623ed4f5
MD5 a8297fe6b15cf693efddb4642895e93e
BLAKE2b-256 401495346644311caed683dcee8b504304f014e0651988110fec82bf4aaf0647

See more details on using hashes here.

File details

Details for the file tensorflow_directml_plugin-0.1.0.dev220928-cp310-cp310-manylinux2010_x86_64.whl.

File metadata

File hashes

Hashes for tensorflow_directml_plugin-0.1.0.dev220928-cp310-cp310-manylinux2010_x86_64.whl
Algorithm Hash digest
SHA256 215275370f2ad6a1b8ceda2a3e5a576ed417b3d7170ca9ff8edad3548e0d766b
MD5 1f020103da3a2701d05e224cf89e4087
BLAKE2b-256 68a1273a66cad6dd8b63d050ee750764f5409f7a4a067cc62746cca250df8008

See more details on using hashes here.

File details

Details for the file tensorflow_directml_plugin-0.1.0.dev220928-cp39-cp39-win_amd64.whl.

File metadata

File hashes

Hashes for tensorflow_directml_plugin-0.1.0.dev220928-cp39-cp39-win_amd64.whl
Algorithm Hash digest
SHA256 3bd2b74e429ed16405bd97fa597a8d691c6c933b1114caa45926704e45e71efd
MD5 b24c2fa9cd3e57ee3e41e05683a82d0d
BLAKE2b-256 d8ad86e841162422c67e0edc656e353f2a2c7869e2f782486ea3a57f1963cf2d

See more details on using hashes here.

File details

Details for the file tensorflow_directml_plugin-0.1.0.dev220928-cp39-cp39-manylinux2010_x86_64.whl.

File metadata

File hashes

Hashes for tensorflow_directml_plugin-0.1.0.dev220928-cp39-cp39-manylinux2010_x86_64.whl
Algorithm Hash digest
SHA256 71c24beb149d85d76461dda324512d56637a6f11ca1cdfee4b0c6ea875dffc68
MD5 05611bb0fea3a88c07e731b0e38c389f
BLAKE2b-256 74e11f26ce937330ea39b1eb6aaf7b46ec754d7fe30016216833f747cac75cd2

See more details on using hashes here.

File details

Details for the file tensorflow_directml_plugin-0.1.0.dev220928-cp38-cp38-win_amd64.whl.

File metadata

File hashes

Hashes for tensorflow_directml_plugin-0.1.0.dev220928-cp38-cp38-win_amd64.whl
Algorithm Hash digest
SHA256 42c532635951318cf4207b2fc7f54079fc32a89c416521de632d7aeb7c7682f7
MD5 c0637081d9785b20b375c55a9b97b1d2
BLAKE2b-256 fdff1edf372d9fb45a805ebc168c790342f23fa489664d6b83fffb9f980c00c4

See more details on using hashes here.

File details

Details for the file tensorflow_directml_plugin-0.1.0.dev220928-cp38-cp38-manylinux2010_x86_64.whl.

File metadata

File hashes

Hashes for tensorflow_directml_plugin-0.1.0.dev220928-cp38-cp38-manylinux2010_x86_64.whl
Algorithm Hash digest
SHA256 2424253c27d163cd93090a02dd46f7771ebaf1ff89b252d244b8f0c8c699bd3d
MD5 10b52636426f65e2bbfcd330a8afc0db
BLAKE2b-256 4685facf57822c43654b81d4346cd2db4a2ed0ff9e5a39f87b0c41756bf4b12f

See more details on using hashes here.

File details

Details for the file tensorflow_directml_plugin-0.1.0.dev220928-cp37-cp37m-win_amd64.whl.

File metadata

File hashes

Hashes for tensorflow_directml_plugin-0.1.0.dev220928-cp37-cp37m-win_amd64.whl
Algorithm Hash digest
SHA256 ade51d79e0d51c1c69767e2557808b309df351905dee60f2fbd723896eb65e5d
MD5 6ea34c1dafcdf361ba6e3afcf60c41a2
BLAKE2b-256 82a9672e5c27c4ab3d8c600c1712348e42fa2e345edd3383df8406129440681c

See more details on using hashes here.

File details

Details for the file tensorflow_directml_plugin-0.1.0.dev220928-cp37-cp37m-manylinux2010_x86_64.whl.

File metadata

File hashes

Hashes for tensorflow_directml_plugin-0.1.0.dev220928-cp37-cp37m-manylinux2010_x86_64.whl
Algorithm Hash digest
SHA256 fd8e38b672d3a968aa47a4ea0500f1ed6669b9077319f8859bde5c7e17ee7803
MD5 f7c1c40ede115473ab09bac00eb4491d
BLAKE2b-256 22232e31927e561241cd8e272b9f21455f82219236ad128682d734e2356747be

See more details on using hashes here.

Supported by

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