reland [ROCm] ubuntu version check in install_rocm.sh (#58164)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58164
Reland of https://github.com/pytorch/builder/pull/764
This reverts commit 6404184700159213f7d64df62537e238822f8b15.
Test Plan: Imported from OSS
Reviewed By: walterddr
Differential Revision: D28387195
Pulled By: seemethere
fbshipit-source-id: 905a14d9ed5e85a7dff7da3d1c4a628320ff7451
