In current ML environment, there's no single thing as "GPU support" as nVidia and AMD are two very different (and unequal) worlds - if you want things to work o

In current ML environment, there's no single thing as "GPU support" as nVidia and AMD are two very different (and unequal) worlds - if you want things to work on AMD, then you should look for things which are explicitly labeled as AMD-compatible, because the default state of "has GPU support" is slang for "works on CUDA, but also might work on AMD if stars are right".