This release is a pre-release and may not be stable for production use.
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.
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
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file tensorflow_directml_plugin-0.3.0.dev221212-cp310-cp310-win_amd64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.3.0.dev221212-cp310-cp310-win_amd64.whl
- Upload date:
- Size: 9.0 MB
- Tags: CPython 3.10, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
05f02b7c77a3ebe5631a871d7a5a285338635453f041065f77e9691363a703b9
|
|
| MD5 |
310dd81785a0feceb90f7744be56d7ab
|
|
| BLAKE2b-256 |
a08921dd039d97cc2d4bf6899cb5bca00109ffcb4b6df7e043dd9fca554ad34e
|
File details
Details for the file tensorflow_directml_plugin-0.3.0.dev221212-cp310-cp310-manylinux2010_x86_64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.3.0.dev221212-cp310-cp310-manylinux2010_x86_64.whl
- Upload date:
- Size: 26.8 MB
- Tags: CPython 3.10, manylinux: glibc 2.12+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
2dec25ac66cb94c7dd161f1cbdb2049956543eab8bbe6b4905ba9f92d70e565f
|
|
| MD5 |
a446eac480107191241bf159e251d008
|
|
| BLAKE2b-256 |
7e1981dd736912f0bf51559e60f243028a3a3ad867ff0ffaecec521628ce2631
|
File details
Details for the file tensorflow_directml_plugin-0.3.0.dev221212-cp39-cp39-win_amd64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.3.0.dev221212-cp39-cp39-win_amd64.whl
- Upload date:
- Size: 9.0 MB
- Tags: CPython 3.9, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
987f19d7d3c779ccafd41988f9cc2cf7b2310dcbfaea4fda6b9fecacf2d39cfe
|
|
| MD5 |
18fd36d3d6eacc37e4524a479d140c63
|
|
| BLAKE2b-256 |
fdef1af84b61572379a318a1ec0b509905eba243ee1fe201ea5cfe34e660e8ad
|
File details
Details for the file tensorflow_directml_plugin-0.3.0.dev221212-cp39-cp39-manylinux2010_x86_64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.3.0.dev221212-cp39-cp39-manylinux2010_x86_64.whl
- Upload date:
- Size: 26.8 MB
- Tags: CPython 3.9, manylinux: glibc 2.12+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b8bc28d842be1259db8de0ac2e0e67c8cd23bba685db78bbed426e239d8655d6
|
|
| MD5 |
217025e2e0cfa81593301888b8fceba8
|
|
| BLAKE2b-256 |
76fc635fdf2f049ab5914c4a86743dabae2f1e34ef0a6f113bf0c4aafe3a80f0
|
File details
Details for the file tensorflow_directml_plugin-0.3.0.dev221212-cp38-cp38-win_amd64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.3.0.dev221212-cp38-cp38-win_amd64.whl
- Upload date:
- Size: 9.0 MB
- Tags: CPython 3.8, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
09d6957864b87abb5b9e64f08b983828a9dc529f5558c751588efab06fbd4dda
|
|
| MD5 |
0af55465100082f74192ac3c9ab8ee4d
|
|
| BLAKE2b-256 |
05669d818f5c9a04acf8dca0ec063fd4e82e2433f5be2c3806648ffcdc53302d
|
File details
Details for the file tensorflow_directml_plugin-0.3.0.dev221212-cp38-cp38-manylinux2010_x86_64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.3.0.dev221212-cp38-cp38-manylinux2010_x86_64.whl
- Upload date:
- Size: 26.8 MB
- Tags: CPython 3.8, manylinux: glibc 2.12+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8219fd84beb2820ed84e9887ce75e81811f9f204b4128f152e86207d9292541b
|
|
| MD5 |
555fb02fcc1f74c1d1eb27ead9561cc6
|
|
| BLAKE2b-256 |
c2c64712695e46c2a118c5f8412a0f09b26a2257cd04690a294ab63835a64d27
|
File details
Details for the file tensorflow_directml_plugin-0.3.0.dev221212-cp37-cp37m-win_amd64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.3.0.dev221212-cp37-cp37m-win_amd64.whl
- Upload date:
- Size: 9.0 MB
- Tags: CPython 3.7m, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
5cb3ac2577a16fdec841fea27cc0341a779a16a99d2a81e2121d0b2e80cb4209
|
|
| MD5 |
47c2af8b95ed8e9b6e4980ac966dd13c
|
|
| BLAKE2b-256 |
726001fd1977699eba9ad1059a823ff424eeb1b05515c887e6d459b7fcb346e8
|
File details
Details for the file tensorflow_directml_plugin-0.3.0.dev221212-cp37-cp37m-manylinux2010_x86_64.whl.
File metadata
- Download URL: tensorflow_directml_plugin-0.3.0.dev221212-cp37-cp37m-manylinux2010_x86_64.whl
- Upload date:
- Size: 26.8 MB
- Tags: CPython 3.7m, manylinux: glibc 2.12+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.7.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1bf15efc698fe06786d60df9152e8ab755226af56137da7684794c32764f2f35
|
|
| MD5 |
d3d17a073c176c94d1d94cde1f043803
|
|
| BLAKE2b-256 |
5a955ccfcaa92d6b01b030a8c0a1ed24e661111cc028053d22b97790ddbf9d7a
|