OSError: libtorch_hip.so: cannot open shared object file in Speaker_Diarization_Inference

Followed the steps here : https://github.com/NVIDIA/NeMo/blob/main/tutorials/speaker_tasks/Speaker_Diarization_Inference.ipynb
to run in jupyter notebook and just ran the whole notebook as it is. But got the error `OSError: libtorch_hip.so: cannot open shared object file: No such file or directory` when importing `nemo.collections.asr.models`


**Environment details**
![image](https://user-images.githubusercontent.com/66141447/177719440-0e6a7cbe-4266-4f69-ae8c-01ebb527b50d.png)

- OS version Ubuntu 20.04
- PyTorch version 1.11.0
- Python version 3.9.12

