How to use PyTorch with AMD VEGA GPU


(Furkan) #1

I have serious problems on this issue. I know, natively, it isn’t supporting yet. However I am observing some “movements” on the AMD side with the new ROCm platform. They have a fork of PyTorch on their github. I wonder if anyone tried to install that port? I would be grateful if someone guide me through the installation and getting it to work.


(Félix Lessange) #2

Pretty interested too