Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Import Error with Torch 1.9.0+cu111 #48

Closed
mb010 opened this issue Sep 10, 2021 · 2 comments
Closed

Import Error with Torch 1.9.0+cu111 #48

mb010 opened this issue Sep 10, 2021 · 2 comments

Comments

@mb010
Copy link

mb010 commented Sep 10, 2021

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
Copy link
Contributor

ejnnr commented Sep 10, 2021

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
Copy link
Collaborator

Gabri95 commented Oct 21, 2021

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants