2 ms·
In the deep learning world every major framework works on top of Cudnn which works on top of CUDA Pytorch, Tensorflow you name it. https://github.com/pytorch/p
by deepGem 6y ago
In the deep learning world every major framework works on top of Cudnn which works on top of CUDA Pytorch, Tensorflow you name it.
https://github.com/pytorch/pytorch/issues/10657 https://github.com/pytorch/pytorch/issues/10657
That is the state of Pytorch support for AMD GPUs.