Feb 25, 2022
Post comments count0
Post likes count0
PyTorch-DirectML: Preview Release 2

A few months ago, we released the first preview of PyTorch-DirectML: a hardware accelerated backend for training PyTorch models on any DirectX12 GPU on Windows and the Windows Subsystem for Linux (WSL). Today, we are releasing the Second Preview with significant performance improvements and greater coverage for computer vision models. PyTorch-Di...