Add check for rocminfo in the installation script to replace the nightly URL with the appropriate ROCm whl. (#1279)

* Add check for rocminfo in the installation script to replace the nightly URL with the appropriate ROCm whl.

* Update install/install_requirements.sh

use nightly

Co-authored-by: Jeff Daily <[email protected]>

---------

Co-authored-by: vmpuri <[email protected]>
Co-authored-by: Jeff Daily <[email protected]>

Author: 3 people

install/install_requirements.sh

@@ -67,6 +67,9 @@ TUNE_NIGHTLY_VERSION=dev20240928
 if [[ -x "$(command -v nvidia-smi)" ]];
 then
   TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cu121"
+elif [[ -x "$(command -v rocminfo)" ]];
+then
+  TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/rocm6.2"
 else
   TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cpu"
 fi