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

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']