Showing results for October 2021 - Windows AI Platform

Oct 21, 2021
Post comments count3
Post likes count0

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...