AMD ROCm on ARCHER2 with PyTorch
I am purposely using an older version of ROCm since the AMD drivers on this HPC cluster is too old. Latest one should also work fine.
Note: This installation is not for regular computers but for HPC clusters or supercomputers.