This is a short guide on how to setup ROCm on Ubuntu. The most important step is that when we install PyTorch we get the latest available ROCm version for it.
- Python 3.10 or newer
- Ubuntu
- Make sure you have the appropriate kernel version, see prerequesites.
Select the desired amdgpu version and download the correct Ubuntu version, replace <version>
and <codename>
. Check the available versions first https://repo.radeon.com/amdgpu-install/
.