ImportError: cannot import name 'container_abcs' from 'torch._six'

[MatCorr]

I am trying to run a basic training script with the ADE20K dataset via the following command:

python tools/train.py configs/SETR/SETR_PUP_DeiT_512x512_160k_ade20k_bs_16_MS.py

And I am running into this error:

ImportError: cannot import name 'container_abcs' from 'torch._six'

Here are the versions of the main components of the environment I am running this on:

• Python 3.10.8
• Torch 1.13.1
• mmcv-full 1.2.2
• mmsegmentation git+https://github.com/fudan-zvg/SETR.git@41162a958abb7851558805558df263a789d0aa98

I am sure this has something to do with version conflicts, but I have tried other combinations and none of them seem to work, usually leading to different errors.

Can anyone shed some light on which versions to use in order to make it work? I am reluctant to downgrade Python and Torch due to GPU-related reasons, but I think I could go as low as Python 3.9 here if necessary.