ImportError under Pytorch 1.5 (nightly)

[jerryzhang-ss]

Hi. I am trying to link warp-ctc to pytorch preview version, but I got this error:

Traceback (most recent call last):
  File "chineseocr/train_ocr.py", line 14, in <module>
    from warpctc_pytorch import CTCLoss
  File "/home/jerry/anaconda3/envs/ocr_1.5/lib/python3.6/site-packages/warpctc_pytorch-0.1-py3.6-linux-x86_64.egg/warpctc_pytorch/__init__.py", line 6, in <module>
    from ._warp_ctc import *
ImportError: /home/jerry/anaconda3/envs/ocr_1.5/lib/python3.6/site-packages/warpctc_pytorch-0.1-py3.6-linux-x86_64.egg/warpctc_pytorch/_warp_ctc.cpython-36m-x86_64-linux-gnu.so: undefined symbol: THPVariableClass

May I know how to solve this? Thanks for your help

[YongjieYang-YY]

The same issue with me. Do you know how to solve this problem?

[Stonesjtu]

Confirmed on pytorch 1.6.0 release.

I think the cpp extension API has some BC breaking changes.

I'm working on a patch on it, so stay tuned.

[Stonesjtu]

@jerryzhang-ss @YongjieYang-YY I solved this problem by re-compiling the pytorch binding after upgrading pytorch. Have you ever tried that before?