[BUG] No module named 'torch._six'

[parthmannan]

Torch does not seem to support torch._six anymore and it has been removed.
Refer - pytorch/pytorch#94709

DeepSpeed still has dependency on it. Example in runtime/utils.py

from torch._six import inf

Torch version to be precise - 2.0.0a0+git5a54537

[weiwangmeta]

Please apply fix " from torch import inf", please let us know if this does not fix your issue.

[styler00dollar]

I cloned git master (8be8c01) now, I did not modify deepspeed, ran into same problem. I used torch==2.0.0.dev20230217+cu118.

Changing to from torch import inf in all files works.

[chintan-donda]

Uninstall the old version of deepspeed, and reinstall it without any version to let it pick up the latest one.

pip uninstall deepspeed
pip install deepspeed

This worked for me.

[sanyanliangyu]

Uninstall the old version of deepspeed, and reinstall it without any version to let it pick up the latest one.

pip uninstall deepspeed
pip install deepspeed

This worked for me.

WORK FOR ME TOO

[mclohrk]

not work for me in google colab