PyTorch-DirectML: Preview Release 3

The Windows AI Platform is excited to announce the Third Preview release of PyTorch-DirectML; a hardware accelerated backend for training PyTorch models on any DirectX12 GPU on Windows and the Windows Subsystem for Linux (WSL).

This preview release includes expanded support for computer vision models from the Torchvision library. On the DirectML GitHub, you’ll find additional samples to help you get starting using image classification and object detection models.

Try it out today! Download from PyPI and leave feedback on the DirectML GitHub.