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.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
File details
Details for the file tensorflow_directml_plugin-0.0.1.dev220621-cp310-cp310-win_amd64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.0.1.dev220621-cp310-cp310-win_amd64.whl
- Upload date:
- Size: 7.9 MB
- Tags: CPython 3.10, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.10.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 43649dc119d35849baf764bcdb06db449b17870f3ff64a3bbce237fae8f13bbe |
|
| MD5 | c4a3a865122468e927439371abeb0e56 |
|
| BLAKE2b-256 | 94dc2ba4545cedc93895a628fa77d3a5cd8f9d7d48a806ce48ea8bca176775d1 |
File details
Details for the file tensorflow_directml_plugin-0.0.1.dev220621-cp310-cp310-manylinux2010_x86_64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.0.1.dev220621-cp310-cp310-manylinux2010_x86_64.whl
- Upload date:
- Size: 24.1 MB
- Tags: CPython 3.10, manylinux: glibc 2.12+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.10.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 1188b477329a265fbe2b4c8375c964dcad18f1049628f94145a9b9ce61a0c6c6 |
|
| MD5 | 079ed6432fb1f979374e51ec1c25f817 |
|
| BLAKE2b-256 | b2eb458d3772ee4683d61401df379675bb1acf0b9cda6a0bcde8de819d766666 |
File details
Details for the file tensorflow_directml_plugin-0.0.1.dev220621-cp39-cp39-win_amd64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.0.1.dev220621-cp39-cp39-win_amd64.whl
- Upload date:
- Size: 7.9 MB
- Tags: CPython 3.9, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.10.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | f9abadb6168fd0a9684c107fa396d039b4c6ea23fe653c017c0ec0b242055e89 |
|
| MD5 | bb4549abe73243e832651bc9f9a38676 |
|
| BLAKE2b-256 | b19b4da693ec9fbb6c0937c4e7dffd26dbfe349cd3e2d9d21a0e81e7304ab640 |
File details
Details for the file tensorflow_directml_plugin-0.0.1.dev220621-cp39-cp39-manylinux2010_x86_64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.0.1.dev220621-cp39-cp39-manylinux2010_x86_64.whl
- Upload date:
- Size: 24.1 MB
- Tags: CPython 3.9, manylinux: glibc 2.12+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.10.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | a016c26bf6858ff4747d4d175eccdd7ba04c2a58ca519ab1c956d16a8f4f0cb4 |
|
| MD5 | 618af8fdde787398800b3ea18d016598 |
|
| BLAKE2b-256 | 08843a56f37427801ed9be0181e848041ab8bc5c7b6e2aaadf6695c0671fa23a |
File details
Details for the file tensorflow_directml_plugin-0.0.1.dev220621-cp38-cp38-win_amd64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.0.1.dev220621-cp38-cp38-win_amd64.whl
- Upload date:
- Size: 7.9 MB
- Tags: CPython 3.8, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.10.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | b74116372964b5d1c77d474070fe592ba42bdd0bc259117fda17cced70a195d0 |
|
| MD5 | d71a5668371ac0ede29f2efe3365736e |
|
| BLAKE2b-256 | ecdfa5471698ca4ac6da1e1f223ff8a0290e228cb12d467120dbe01bcd8d65b0 |
File details
Details for the file tensorflow_directml_plugin-0.0.1.dev220621-cp38-cp38-manylinux2010_x86_64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.0.1.dev220621-cp38-cp38-manylinux2010_x86_64.whl
- Upload date:
- Size: 24.1 MB
- Tags: CPython 3.8, manylinux: glibc 2.12+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.10.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | a63cf3ee93d2c31b0d195ae98fa4c84971b0d714e87d5f67aaaf1c822465c7ab |
|
| MD5 | 6b9aa358415e33f0c3872719ae39bc33 |
|
| BLAKE2b-256 | 840cf082853fc6285134fc24cc19a51bb41ed38a853c1f00f08cac150cdea201 |
File details
Details for the file tensorflow_directml_plugin-0.0.1.dev220621-cp37-cp37m-win_amd64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.0.1.dev220621-cp37-cp37m-win_amd64.whl
- Upload date:
- Size: 7.9 MB
- Tags: CPython 3.7m, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.10.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | e40f67dffcd4c9d32e1cb38418e917c9bc63fd3b7bbdeb54123079f311e384c7 |
|
| MD5 | 48681bd6fe94b7024c892ca80712d34c |
|
| BLAKE2b-256 | ff2951c0d472e84c8325f1fa02785026b11c3f3af2749889f8f3317404a1486a |
File details
Details for the file tensorflow_directml_plugin-0.0.1.dev220621-cp37-cp37m-manylinux2010_x86_64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.0.1.dev220621-cp37-cp37m-manylinux2010_x86_64.whl
- Upload date:
- Size: 24.1 MB
- Tags: CPython 3.7m, manylinux: glibc 2.12+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.10.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 5eee657664a83db007b350e7953e60cdd90ffe6ff59f544378b640b4d9d36ade |
|
| MD5 | a2d76d14ae537fc49d1c6e7b36db4b50 |
|
| BLAKE2b-256 | 43a4490ff11fbbc7b4a6aaf63c0003621f60d6ac9c16ba952c090adcb06799f3 |
