[kaelinl]

I haven't used it personally, but my understanding is that AMD/ROCm-accelerated backends for PyTorch overload the "cuda" device and module so that ROCm shows up as CUDA for feature testing. They want to make a transition seamless, and lots of existing code checks for CUDA, so they do what's necessary to make that existing code run.

[mappu]

That seems to be the case for me out-of-the-box.

    $ python3 -m venv .
    $ ./bin/pip3 install torch --index-url https://download.pytorch.org/whl/rocm5.6
    $ ./bin/python3 -c 'import torch; print([torch.cuda.is_available(), torch.cuda.get_device_name(0)])'
    [True, 'AMD Radeon RX 6600 XT']

[bragr]

Which tools actually pull down those packages though? If anything that's a counter point to the Intel CEO: industry is adopting CUDA as the defacto standard.