1ytic / warp-rnnt

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

RuntimeError: rnnt_loss status 1 #22

Closed CLC2019 closed 2 years ago

CLC2019 commented 2 years ago

Hello,the error occurs when I run my project.Then I go to run 'python -m warp_rnnt.test',but the same error occurs which is showed as follows. Could you give me some ideas to solve it?The pytorch version is 1.6.0.The CUDA version is 10.1.The gcc version is 7.5.0. image

1ytic commented 2 years ago

Has it worked successfully before? Maybe you changed the gpu card? Try to reinstall the cuda driver.

CLC2019 commented 2 years ago

@1ytic I am really sorry for replying late.It's my first test on my server.During this time,I tested it in other servers without errors.And I also reinstalled the cuda driver with the same error on my sersver,The pytorch version is 1.6.0.The CUDA version is 10.0.The cudatoolkit version is 10.1.Could you give me some advice?Thanks a lot.

1ytic commented 2 years ago

Try to recompile the warp-rnnt package.

CLC2019 commented 2 years ago

@1ytic Thanks a lot.The error is solved.

jiay7 commented 2 years ago

@1ytic Thanks a lot.The error is solved.

Hi, I got the same problem. How did you solve it? python=3.8 torch verison=1.10.2 cudatoolkit=10.2.89 CUDA version=10.2