Autodetect ROCM version for torch installs