Theano using alot of CPU despite GPU enabled

Viewed 11

I am running theano with the config set to cuda in the device=cuda. However, the amount of cpu used is alarming, almost 3000%. CPU usage.

my theanorc setup is as such:

mode = Fast run
OMP_NUM_THREADS = 5
optimizer = fast run
gpuarray.preallocate=0.8
fast_math = True

Is there any theano flag that allows more computation on gpu rather than cpu?

0 Answers
Related