2 ms·
yes that was my conclusion after some research on this. there is an open issue on github for AMD support on pytorch, and looks like something works on arch linu
by d0mdo0ss 6y ago
yes that was my conclusion after some research on this. there is an open issue on github for AMD support on pytorch, and looks like something works on arch linux, but really sounds like support is still in the hacking stage and far from production mode.
https://github.com/pytorch/pytorch/issues/10657 https://github.com/pytorch/pytorch/issues/10657
- t-vi 6y agoSupport for PyTorch on ROCm is fairly good. I have built it from the dev branch without much trouble for all year. There has been ROCm CI even longer, my patch to print ROCm system information for bug reports was merged last week. If you know where to look (i.e. it's public but unannounced) you can see that nightly wheels have been built for the last few days. So I would expect that some time between now and the Developer Day in November we'll see ROCm appear on PyTorch's "get started" page.