Error in importing pytorch_lightening

[JVedant]

Hi, I'm using torch version '1.9.0.dev20210601+cu113' and while importing pytorch_lightening, it's displaying an error as follows

ImportError: cannot import name 'container_abcs' from 'torch._six' (miniconda3/envs/torchsm_86/lib/python3.8/site-packages/torch/_six.py)
---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-4-b2cf6d7c113b> in <module>
----> 1 import pytorch_lightning as pl
      2 import torch.nn as nn
      3 from torch.utils.data import Dataset, DataLoader
      4 import torch
      5 from torchvision import datasets

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/__init__.py in <module>
     18 _PROJECT_ROOT = os.path.dirname(_PACKAGE_ROOT)
     19 
---> 20 from pytorch_lightning import metrics  # noqa: E402
     21 from pytorch_lightning.callbacks import Callback  # noqa: E402
     22 from pytorch_lightning.core import LightningDataModule, LightningModule  # noqa: E402

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/metrics/__init__.py in <module>
     13 # limitations under the License.
     14 
---> 15 from pytorch_lightning.metrics.classification import (  # noqa: F401
     16     Accuracy,
     17     AUC,

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/metrics/classification/__init__.py in <module>
     12 # See the License for the specific language governing permissions and
     13 # limitations under the License.
---> 14 from pytorch_lightning.metrics.classification.accuracy import Accuracy  # noqa: F401
     15 from pytorch_lightning.metrics.classification.auc import AUC  # noqa: F401
     16 from pytorch_lightning.metrics.classification.auroc import AUROC  # noqa: F401

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/metrics/classification/accuracy.py in <module>
     16 from torchmetrics import Accuracy as _Accuracy
     17 
---> 18 from pytorch_lightning.metrics.utils import deprecated_metrics, void
     19 
     20 

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/metrics/utils.py in <module>
     27 from torchmetrics.utilities.distributed import reduce as _reduce
     28 
---> 29 from pytorch_lightning.utilities import rank_zero_deprecation
     30 from pytorch_lightning.utilities.imports import _TORCHMETRICS_GREATER_EQUAL_0_3, _TORCHMETRICS_LOWER_THAN_0_3
     31 

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/utilities/__init__.py in <module>
     16 import numpy
     17 
---> 18 from pytorch_lightning.utilities.apply_func import move_data_to_device  # noqa: F401
     19 from pytorch_lightning.utilities.distributed import AllGatherGrad, rank_zero_info, rank_zero_only  # noqa: F401
     20 from pytorch_lightning.utilities.enums import (  # noqa: F401

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/utilities/apply_func.py in <module>
     25 
     26 from pytorch_lightning.utilities.exceptions import MisconfigurationException
---> 27 from pytorch_lightning.utilities.imports import _compare_version, _TORCHTEXT_AVAILABLE
     28 
     29 if _TORCHTEXT_AVAILABLE:

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/utilities/imports.py in <module>
     70 _TORCH_GREATER_EQUAL_1_9 = _compare_version("torch", operator.ge, "1.9.0")
     71 
---> 72 _APEX_AVAILABLE = _module_available("apex.amp")
     73 _BOLTS_AVAILABLE = _module_available("pl_bolts")
     74 _DEEPSPEED_AVAILABLE = _module_available("deepspeed")

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/pytorch_lightning/utilities/imports.py in _module_available(module_path)
     35     """
     36     try:
---> 37         return find_spec(module_path) is not None
     38     except AttributeError:
     39         # Python 3.6

~/miniconda3/envs/torchsm_86/lib/python3.8/importlib/util.py in find_spec(name, package)
     92         parent_name = fullname.rpartition('.')[0]
     93         if parent_name:
---> 94             parent = __import__(parent_name, fromlist=['__path__'])
     95             try:
     96                 parent_path = parent.__path__

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/apex/__init__.py in <module>
      6     from . import parallel
      7 
----> 8 from . import amp
      9 from . import fp16_utils
     10 

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/apex/amp/__init__.py in <module>
----> 1 from .amp import init, half_function, float_function, promote_function,\
      2     register_half_function, register_float_function, register_promote_function
      3 from .handle import scale_loss, disable_casts
      4 from .frontend import initialize, state_dict, load_state_dict
      5 from ._amp_state import master_params, _amp_state

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/apex/amp/amp.py in <module>
----> 1 from . import compat, rnn_compat, utils, wrap
      2 from .handle import AmpHandle, NoOpHandle
      3 from .lists import functional_overrides, torch_overrides, tensor_overrides
      4 from ._amp_state import _amp_state
      5 from .frontend import *

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/apex/amp/rnn_compat.py in <module>
----> 1 from . import utils, wrap
      2 
      3 import torch
      4 _VF = torch._C._VariableFunctions
      5 RNN_NAMES = ['rnn_relu', 'rnn_tanh', 'gru', 'lstm']

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/apex/amp/wrap.py in <module>
      1 from . import compat
      2 from . import utils
----> 3 from ._amp_state import _amp_state
      4 from . import rnn_compat
      5 

~/miniconda3/envs/torchsm_86/lib/python3.8/site-packages/apex/amp/_amp_state.py in <module>
     12     import collections.abc as container_abcs
     13 else:
---> 14     from torch._six import container_abcs
     15 
     16 

ImportError: cannot import name 'container_abcs' from 'torch._six' (miniconda3/envs/torchsm_86/lib/python3.8/site-packages/torch/_six.py)

what can be done to solve/overcome this error ?

Note: I'm using the nightly build of PyTorch as the stable version doesn't support Cuda 11.2

[JVedant]

This got solved by adding import collections.abc as container_abcs in
miniconda3/envs/environment_name/lib/python3.8/site-packages/torch/_six.py