Preview 2 of the DirectML PluggableDevice for TensorFlow is available!
You can pick up the new package on PyPI today.
Highlights include:
Upgrade to DirectML version 1.9.1- Added dependency on
tensorflow-cpu>=2.10.0 - Overall improvements in memory allocation and performance
Read the complete list of changes on GitHub
Leave any questions, suggestions, or issues on GitHub. Our team is constantly engaging with the community and would love to hear your input!