Feb 25, 2022
0
0
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...