Hironsan / anago

Bidirectional LSTM-CRF and ELMo for Named-Entity Recognition, Part-of-Speech Tagging and so on.
https://anago.herokuapp.com/
MIT License
1.48k stars 368 forks source link

GPU #29

Closed jodevak closed 6 years ago

jodevak commented 6 years ago

I ran a test across a GPU based instance, but i didnt see any speed improvments. Is there something that i would need to modify in order to support GPU ? thanks

ismaeIfm commented 6 years ago

do you have tensorflow-gpu installed?

jodevak commented 6 years ago

Yes, its an aws instance.

ismaeIfm commented 6 years ago

In order to help you, you need to give more feedback. I would suggest to check that Keras is using tensorflow as backend, that tensorflow-gpu is installed and that cuda and cudnn are properly installed.