[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.