using gpu error:"'module' object has no attribute 'gpu_ctc'"

meijieru / crnn.pytorch

Convolutional recurrent network in pytorch

MIT License

2.39k stars 658 forks source link

using gpu error:"'module' object has no attribute 'gpu_ctc'" #36

Closed lg12170226 closed 7 years ago

lg12170226 commented 7 years ago

run crnn_main.py using --cuda,and then error. Error content：loss_func = warp_ctc.gpu_ctc if is_cuda else warp_ctc.cpu_ctc AttributeError:'module' object has no attribute 'gpu_ctc'. How to deal with?

meijieru commented 7 years ago

Please ensure the warp_ctc is built with gpu support.

lg12170226 commented 7 years ago

@meijieru thank u for your help . i have solved the problem

rohun-tripathi commented 6 years ago

@lg12170226 How did you solve the problem? I get the exact same issue

lg12170226 commented 6 years ago

@rohun-tripathi sorry for long time. Before installing pytorch_binding, make sure to set CUDA_HOME, compile and install will not complain, but when calling gpu wrap_ctc there will be no gpu attribute error