You wouldn't be able to use the PyTorch as built and provided by Ubuntu, as in 'apt-get install python3-torch' [1], as that version currently only has CPU suppo

You wouldn't be able to use the PyTorch as built and provided by Ubuntu, as in 'apt-get install python3-torch' [1], as that version currently only has CPU support. You could still use a version provided by a third party, for example directly from pytorch.org, or a containerized version provided by ROCm upstream [2]. [1] https://packages.ubuntu.com/mantic/python3-torch https://packages.ubuntu.com/mantic/python3-torch [2] https://hub.docker.com/r/rocm/pytorch https://hub.docker.com/r/rocm/pytorch