How can I train with my GPU?

tensorlayer / seq2seq-chatbot

Chatbot in 200 lines of code using TensorLayer

https://github.com/tensorlayer/tensorlayer

836 stars 316 forks source link

Open cavin12400 opened 5 years ago

cavin12400 commented 5 years ago

Hello, I'm planning to train this with my GPU to make the training faster. The problem is it only utilizes my CPU. Thank you.

pskrunner14 commented 5 years ago

@cavin12400 It should be running on GPU by default. Please check your tensorflow-gpu installation.