Now u can run PyTorch on any DirectX12 supported GPU without a | Machine Learning World

Now u can run PyTorch on any DirectX12 supported GPU without any mess with CUDA(AMD can be used)

https://devblogs.microsoft.com/windowsai/introducing-pytorch-directml-train-your-machine-learning-models-on-any-gpu/