No module named 'torch.utils._device'

[nikuson]

No module named 'torch.utils._device'

[carmocca]

Can you share the complete error stacktrace?

Are you using torch==2.0?

[nikuson]

yes i am using torch 2.0

Can you share the complete error stacktrace?

Are you using torch==2.0?

yes i am using torch 2.0

[nikuson]

Can you share the complete error stacktrace?

Are you using torch==2.0?

Traceback (most recent call last): File "D:\Python\Llama\llama\generate.py", line 10, in
from lit_llama.utils import EmptyInitOnDevice File "D:\Python\Llama\llama\lit_llama\utils.py", line 8, in import torch.utils._device ModuleNotFoundError: No module named 'torch.utils._device'

[carmocca]

Can you share the output of

pip list | grep torch and python -c 'import torch; print(torch.__version__)'?

You might have non-release version that doesn't include that file. Reinstalling torch by following the instructions at www.pytorch.org/get-started/locally should help

[Ramakrishna176420]

Traceback (most recent call last): File "sketch3.py", line 1, in import torch File "/home/stud2201/.local/lib/python3.8/site-packages/torch/init.py", line 21, in from ._utils import _import_dotted_name ModuleNotFoundError: No module named 'torch._utils'