Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

pytorch_lightning Trainer doesn't recognize hpu #10

Open
jingkang99 opened this issue Mar 27, 2024 · 0 comments
Open

pytorch_lightning Trainer doesn't recognize hpu #10

jingkang99 opened this issue Mar 27, 2024 · 0 comments

Comments

@jingkang99
Copy link

gaudi2 system with fresh pytorch installation
pip list | grep -P "torch|torchvision|torchmetrics|protobuf"
image

Basic Pytorch training is fine - followed steps on https://docs.habana.ai/en/latest/PyTorch/Getting_Started_with_PyTorch_and_Gaudi/Getting_Started_with_PyTorch.html

from pytorch_lightning import LightningModule, Trainer

Try following pytorch-lightening hello world example, get some error
https://github.com/HabanaAI/Gaudi-tutorials/blob/main/Lightning/Introduction/mnist-hello-world.ipynb

but "hpu" not recognized
image

what should be installed?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant