Traning NLU on GPU


While training NLU it is consuming 100% CPU with following specs.

8 GB RAM i5 7th gen 2 GB Nvidia 940 mx

I need help regarding setting up the GPU for training.

It would help us debug if you gave us detailed information about the steps that you took as well as your operating system.

One thing to point out though. While DIET is able to use the GPU via tensorflow, the other components in your NLU pipeline cannot. A countvectorizer cannot run faster when there’s a GPU around and will only use a CPU. That said, the CV object is made with scikit-learn so that can only use 1 CPU core at a time. So I’m curious what else might be causing it.

Hey thanks for the response but, I am not trying to set up the GPU anymore.