3 ms·

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

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.

bragr 3y ago
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.
mappu 3y ago
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']