GPU is not working.

[gregcho]

Hello.
I finally find good project for me.
but I have 1 issue using GPU, CPU version is fine.

when I load program like this.

    read = easyocr.Reader(['en'])
    text = read.readtext('D:/workspace/EasyOCR-master/examples/4.jpg')
    print(text)

I got this message :
RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu

[gregcho]

I have two GPU.

• GeForce GTX 1070 Ti
• pytorch version : 1.5.1

[rkcosmos]

Can you try specify which gpu to be used by using gpu = 'your gpu' argument? See this ssue.

[gregcho]

'your gpu' argument means gpu number?

[rkcosmos]

Something like 'cuda:0' or 'cuda:1'. See this discussion on pytorch.

[gregcho]

thanks!