tensorflow-directml-plugin

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

Release history Release notifications | RSS feed

This version

0.4.0.dev230202 pre-release

Feb 3, 2023

0.3.0.dev221212 pre-release

Dec 13, 2022

0.2.0.dev221020 pre-release

Oct 21, 2022

0.1.1.dev221004 pre-release

Oct 5, 2022

0.1.0.dev220928 pre-release

Sep 29, 2022

0.0.1.dev220621 pre-release

Jun 22, 2022

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.4.0.dev230202-cp310-cp310-win_amd64.whl (9.0 MB view hashes)

Uploaded Feb 3, 2023 cp310

tensorflow_directml_plugin-0.4.0.dev230202-cp310-cp310-manylinux2010_x86_64.whl (26.8 MB view hashes)

Uploaded Feb 3, 2023 cp310

tensorflow_directml_plugin-0.4.0.dev230202-cp39-cp39-win_amd64.whl (9.0 MB view hashes)

Uploaded Feb 3, 2023 cp39

tensorflow_directml_plugin-0.4.0.dev230202-cp39-cp39-manylinux2010_x86_64.whl (26.8 MB view hashes)

Uploaded Feb 3, 2023 cp39

tensorflow_directml_plugin-0.4.0.dev230202-cp38-cp38-win_amd64.whl (9.0 MB view hashes)

Uploaded Feb 3, 2023 cp38

tensorflow_directml_plugin-0.4.0.dev230202-cp38-cp38-manylinux2010_x86_64.whl (26.8 MB view hashes)

Uploaded Feb 3, 2023 cp38

tensorflow_directml_plugin-0.4.0.dev230202-cp37-cp37m-win_amd64.whl (9.0 MB view hashes)

Uploaded Feb 3, 2023 cp37

tensorflow_directml_plugin-0.4.0.dev230202-cp37-cp37m-manylinux2010_x86_64.whl (26.8 MB view hashes)

Uploaded Feb 3, 2023 cp37

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp310-cp310-win_amd64.whl

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp310-cp310-win_amd64.whl
Algorithm	Hash digest
SHA256	`77e9c15ed2d3b20f55ad15a7e2354c0ef7843aad2b09dfa979d8dd753b76154e`
MD5	`626872b6babb4a738d0b5f27ee4f8925`
BLAKE2b-256	`383913c9b632bbed15e5bb4aa65e871d6b804fcea5747e97fc82d9b14353b895`

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp310-cp310-manylinux2010_x86_64.whl

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp310-cp310-manylinux2010_x86_64.whl
Algorithm	Hash digest
SHA256	`ecaf259ee58d43724f51eba09d498e8c114567aed2684072ee276b5d1c6a321b`
MD5	`99ead696f03bc62a1c536574b7e4f20c`
BLAKE2b-256	`8b3f843d6913df494510bdbf1e38a91b4acd0b4df58a8efe9032cdb553f4e235`

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp39-cp39-win_amd64.whl

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp39-cp39-win_amd64.whl
Algorithm	Hash digest
SHA256	`b2e881442531f79a9004e160f8983d2b643c9cc8de888c409de1ffca24222933`
MD5	`691c2e5a59de21a6f743ab60d3329b7a`
BLAKE2b-256	`24d55a7334a2321c4dea3946f1e94987d6e8641ec9f56c5912aef017d3de383e`

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp39-cp39-manylinux2010_x86_64.whl

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp39-cp39-manylinux2010_x86_64.whl
Algorithm	Hash digest
SHA256	`3333fa1f9071cd652d231b066cf4c557284c336778db178f7eb4bc31fdeec426`
MD5	`553c7c5f7a9fa33f71e08269716ac55f`
BLAKE2b-256	`81b3228fa82627aa0661f6fca955defdd1f7dfd78d97a2d33ed6f4f487ba64ce`

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp38-cp38-win_amd64.whl

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp38-cp38-win_amd64.whl
Algorithm	Hash digest
SHA256	`e295a9f3fe7e2d43034ab6cd7eea778f9e8646b4500328f0935bb89ba1a46a35`
MD5	`1df54f63d1bb96cf5c96c9bd03cf92f5`
BLAKE2b-256	`ecdf2ed802e1920c169db338d19aee8a9e1e76a2387e7755684aac0f98e0708d`

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp38-cp38-manylinux2010_x86_64.whl

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp38-cp38-manylinux2010_x86_64.whl
Algorithm	Hash digest
SHA256	`d92ff5a7c557679e93a0f5d1f257a46919a0fffe3cbcbf9234681f547e50747a`
MD5	`14e9fdb546b87cdcda943aa96f80ea53`
BLAKE2b-256	`37615350ccc04431563f04ee2a995e06490a56aa29ee601d85d6fc76fac074aa`

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp37-cp37m-win_amd64.whl

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp37-cp37m-win_amd64.whl
Algorithm	Hash digest
SHA256	`daf28deb4cfbc7e2a49acdd01a873e972977079078c0be5b021414b547cc34e0`
MD5	`cc1f4f6a83d9664f52f412c5c0336e20`
BLAKE2b-256	`7cdaf5888b4308f55ba6ddccebf01cce583a81ae1de342ba6ef5eb478c53c3fd`

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp37-cp37m-manylinux2010_x86_64.whl

Hashes for tensorflow_directml_plugin-0.4.0.dev230202-cp37-cp37m-manylinux2010_x86_64.whl
Algorithm	Hash digest
SHA256	`2bf2d3d229de0a1c3c755e8d9fdca9a44b003a0729fc639c02cbca91448e0b4e`
MD5	`afa0fe9de8ab458e1227fddf1c56400c`
BLAKE2b-256	`86760202aef1de075be25e3b3f3dd67cb9715e7819c1de1f45f9bf4863420bbd`