How to fix ModuleNotFoundError: No module named 'habana_frameworks.torch'

Lightning-Universe / lightning-transformers

Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

https://lightning-transformers.readthedocs.io

Apache License 2.0

607 stars 77 forks source link

How to fix ModuleNotFoundError: No module named 'habana_frameworks.torch' #282

Closed zolekode closed 1 year ago

zolekode commented 1 year ago

❓ Questions and Help

I am getting this error after following the instructions on installing habana for pytorch and running my script.

OS: Ubuntu 20.04
Packaging: Pip Anyone has an idea on how to resolve it?

Borda commented 1 year ago

Hi, I believe you should be using the Habana docker image where all needed libraries shall be already prepared for you: vault.habana.ai/gaudi-docker/1.5.0/ubuntu20.04/habanalabs/pytorch-installer-1.11.0:latest