There are different pytorch install commands for cuda/ROCm. Usually the key to getting ROCm to work with AI GitHub projects is to edit a config file/launch option and replace the pytorch command with a ROCm one