Showing results for October 2021 - Windows AI Platform

Oct 21, 2021
3
0

Introducing PyTorch-DirectML: Train your machine learning models on any GPU

Adele Parsons
Adele Parsons

The Windows AI team is excited to announce the first preview of DirectML as a backend to PyTorch for training ML models! This release is our first step towards unlocking accelerated machine learning training for PyTorch on any DirectX12 GPU on Windows and the Windows Subsystem for Linux (WSL). In order for you to take advantage of DirectML withi...