seq2seq-chatbot icon indicating copy to clipboard operation
seq2seq-chatbot copied to clipboard

Published 20 hours ago • verified publisher icon tensorlayer

How can I train with my GPU?

Open cavin12400 opened this issue 6 years ago • 1 comments

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.

cavin12400 avatar Dec 02 '18 07:12 cavin12400

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

pskrunner14 avatar Dec 02 '18 07:12 pskrunner14