Train your machine learning models on any GPU with TensorFlow-DirectML
Train your machine learning models on any GPU with TensorFlow-DirectML
“Setting up TensorFlow-DirectML to work with your GPU is as easy as running “pip install tensorflow-directml” in your Python environment of choice. Once TensorFlow-DirectML is installed, it works seamlessly with existing model training scripts.
We assembled a wide range of model scripts from existing TensorFlow tutorials, online learning courses, the TensorFlow Benchmark set, and AI-Benchmark, as well as other commonly used neural networks. This model set provided a breadth of scenarios for ensuring TensorFlow-DirectML has the operator coverage and performance needed for students’ and professionals’ success.
Our team acted on customer feedback throughout the preview, improving the TensorFlow-DirectML experience on Windows and within WSL. Including optimizing specific operators like convolution and batch normalization and fine-tuning GPU scheduling and memory management so TensorFlow gets the most out of DirectML…”