DirectML support for TensorFlow 1.15 is experimental and currently available as a Public Preview.
TensorFlow with DirectML enables training and inference of complex machine learning models on a wide range of DirectX 12-compatible hardware.
TensorFlow with DirectML is supported on both the latest versions of Windows 10 and the Windows Subsystem for Linux.
For detailed instructions on getting started, see GPU accelerated ML training (docs.microsoft.com).
Two sample models are included in this repo to help you get started. These samples include both inference and training scripts, and you can either train the models from scratch or use the supplied pre-trained weights.
For comments, questions, feedback, or if you're having problems, please file an issue or contact us directly at askdirectml@microsoft.com