AMD ROCm GPU Support
Discussions center on ROCm compatibility for AMD GPUs in machine learning frameworks like PyTorch, including user experiences with specific models, supported hardware lists, and comparisons to NVIDIA CUDA.
Activity Over Time
Top Contributors
Keywords
Sample Comments
Will this work with ROCm instead of CUDA?
AMD supports PyTorch out of the box , these comments make me feel no one has tried or even working on this stuff
Replying "ROCm doesn't support your GPU model" in GitHub issues
Should work fine. They recommend AMD gpus.
It works on my 6900XT with ROCm
No worries, for future reference you can check here (hopefully that page will report improved support in the future)https://github.com/RadeonOpenCompute/ROCm#supported-gpus
ROCm works fine on my 2016 Vega Frontier edition, for what it's worth.
I tried the example provided but it is not working. Are AMD GPUs supported?
Probably - but the answer is to avoid ROCM, not pytorch.
How's support for AMD GPUs? I only saw Nvidia listed.