PyTorch 2.0 work using upstream TOT showed the title error, which is ultimately from this PR https://github.com/pytorch/pytorch/pull/94709. File `apex/amp/_initialize.py` `❱ from torch._six import string_classes` Solution is to cherry pick upstream apex for the fix. https://github.com/NVIDIA/apex/commit/6943fd26e04c59327de32592cf5af68be8f5c44e
PyTorch 2.0 work using upstream TOT showed the title error, which is ultimately from this PR pytorch/pytorch#94709.
File
apex/amp/_initialize.py❱ from torch._six import string_classesSolution is to cherry pick upstream apex for the fix.
NVIDIA@6943fd2