kakaobrain/torchlars

mportError: /home/vladimir/anaconda3/lib/python3.7/site-packages/torchlars/_adaptive_lr.cpython-37m-x86_64-linux-gnu.so: undefined symbol: THPVariableClass

ternaus opened this issue · 3 comments

 from torchlars import LARS
  File "/home/vladimir/anaconda3/lib/python3.7/site-packages/torchlars/__init__.py", line 2, in <module>
    from torchlars.lars import LARS
  File "/home/vladimir/anaconda3/lib/python3.7/site-packages/torchlars/lars.py", line 6, in <module>
    from torchlars._adaptive_lr import compute_adaptive_lr
ImportError: /home/vladimir/anaconda3/lib/python3.7/site-packages/torchlars/_adaptive_lr.cpython-37m-x86_64-linux-gnu.so: undefined symbol: THPVariableClass

I have the same problem. Pytorch 1.6.
It may be related to: pytorch/pytorch#38122
Here they are suggesting a solution, but torch is already imported in torchlars before: pytorch/pytorch#6097

I am also having this problem. Have there been any advancements to address this? What are the environment specs for people who have it working?

I am getting this issue with python3.6 + PyTorch 1.4. Also testing with PyTorch 1.6 in hopes that solves it.

I figured out how to solve it. You want to use PyTorch 1.6.0+ (I also used python3.6)

Hope this helps!