4 ms·

> (Site note: our own PyTorch does not yet ROCm support, we're still working on a few dependencies.) Can I clarify something. If we install a stable release of

> (Site note: our own PyTorch does not yet ROCm support, we're still working on a few dependencies.) Can I clarify something. If we install a stable release of Ubuntu now, we wouldn't be able to get PyTorch to work with a 7900 XT/XTX? At least not until you've worked your magic and that's made its way to Ubuntu in a few months?

ckastner 3y ago
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