Traceback (most recent call last):
  File "/mnt/fs/user/llama/panhw/Megatron-LM-main/pretrain_gpt.py", line 5, in <module>
    import torch
  File "/usr/local/lib/python3.10/site-packages/torch/__init__.py", line 237, in <module>
    from torch._C import *  # noqa: F403
ImportError: libgalaxyhip.so.5: cannot open shared object file: No such file or directory