search

1ytic / warp-rnnt

CUDA-Warp RNN-Transducer
MIT License
211 stars 41 forks source link

Not support Pytorch 1.5 #8

Closed li563042811 closed 3 years ago

li563042811 commented 3 years ago

I used warp_rnnt 0.4.0 and tried 0.3.0. Both of them could not work with pytorch 1.5. The traceback showed this:

_C.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN3c1012CUDATensorIdEv

But when I changed pytorch from 1.5 to 1.0. Both of them worked. Here is my env: python 3.7.7 pytorch 1.5 gcc 4.8.5

Maybe you can fix it and your work will be appreciated.

1ytic commented 3 years ago

You probably need to install a proper CUDA libraries. Have a look at this issue #7