[BUG] can't install with ROCm

It happened just during pip install for deepspeed-0.6.5:

  × python setup.py egg_info did not run successfully.
  │ exit code: 1
  ╰─> [8 lines of output]
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "/tmp/pip-install-ruwdavjo/deepspeed_caadfe808785436583e97206db673e2c/setup.py", line 39, in <module>
          rocm_version = OpBuilder.installed_rocm_version()
        File "/tmp/pip-install-ruwdavjo/deepspeed_caadfe808785436583e97206db673e2c/op_builder/builder.py", line 192, in installed_rocm_version
          with open('/opt/rocm/.info/version-dev', 'r') as file:
      FileNotFoundError: [Errno 2] No such file or directory: '/opt/rocm/.info/version-dev'
      [end of output]
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed

I have ROCm but it is installed into the system, not into /opt/rocm and I have no version-dev file. ROCm PyTorch is working fine. It’s Gentoo Linux.

[mrwyattii]

Thanks for testing that for me. I’ll get this PR merged

[stalkerg]

Also your way in PR looks like working fine:

>>> import torch
>>> torch.__version__.split("rocm")[1]
'5.1.1'