ctc use of gpu

use of gpu_ctc

Open saseptim opened this issue 8 years ago • 3 comments

I see that the code seems to be using only cpu_ctc.

How can I get this to use the gpu?

Thanks Avi

Jul 14 '16 09:07 saseptim

@sherjilozair As far as I can see, cpu_ctc_th should work with GPU as well. Is this correct?

Feb 06 '17 07:02 pranv

+1 i think @sherjilozair should really confirm this.

I've tried running rnnctc.py example with theano on GPU mode and saw ~30MB increase in GPU memory usage. Will try bigger network to confirm.

May 23 '17 08:05 dieka13

It only use CPU: https://github.com/sherjilozair/ctc/blob/master/src/ctc_entrypoint.cpp#L181

Aug 14 '17 11:08 reith