Import Error with Torch 1.9.0+cu111

[mb010]

Error while import with Pytorch version 1.9.0+cu111 (maybe earlier)

----> 6 from e2cnn import nn as e2nn
      7 
      8 import utils

/venv/lib/python3.8/site-packages/e2cnn/nn/__init__.py in <module>
      3 from .geometric_tensor import GeometricTensor, tensor_directsum
      4 
----> 5 from .modules import *
      6 
      7 

/venv/lib/python3.8/site-packages/e2cnn/nn/modules/__init__.py in <module>
     48 
     49 from .sequential_module import SequentialModule
---> 50 from .module_list import ModuleList
     51 from .identity_module import IdentityModule
     52 

/venv/lib/python3.8/site-packages/e2cnn/nn/modules/module_list.py in <module>
      2 from .equivariant_module import EquivariantModule
      3 
----> 4 from torch._six import container_abcs
      5 
      6 import torch

ImportError: cannot import name 'container_abcs' from 'torch._six' (/home/s51972mb/Documents/EquivariantSelfAttention/venv/lib/python3.8/site-packages/torch/_six.py)

Potential fix:
NVIDIA/apex#1049 (comment)

[ejnnr]

This should be fixed by #44 (which is not yet on PyPI I think). In the meantime, you can install the latest version directly from Github with

pip install git+https://github.com/QUVA-Lab/e2cnn

[Gabri95]

hey!

Sorry for the delay: PyPI is now also updated with the latest version of the library so this should not be a problem anymore.

Thanks for reporting it

Best,
Gabriele