3 ms·

I haven't tried it myself, but pytorch has supported amd gpus via rocm since 2021. https://www.amd.com/en/technologies/infinity-hub/pytorch https://www.amd.com/

by eyegor 4y ago

I haven't tried it myself, but pytorch has supported amd gpus via rocm since 2021. https://www.amd.com/en/technologies/infinity-hub/pytorch https://www.amd.com/en/technologies/infinity-hub/pytorch