CODEWITHSUNDEEP
theanolasagnetheano-cuda
yazfield
yazfield

Reputation: 1283

Theano falls back to CPU

I am training a model in Theano 0.9 and Lasagne 0.1 and want to run it on GPU. I've set THEANO_FLAGS as follows:

THEANO_FLAGS=device=gpu0,force_device=True,floatX=float64

Theano prints it is using GPU

Using gpu device 0: GeForce GTX 980 Ti (CNMeM is disabled, cuDNN 4007)

However, I noticed it's not, profiling shows that it's using CorrMM operation which is according to the docs

CorrMM This is a CPU-only 2d correlation implementation taken from caffe’s cpp implementation and also used by Torch.

I have CUDA Toolkit 7.5 installed, Tensorflow works perfectly on GPU. For some reason Theano is falling back to CPU, it is supposed to cause an error due to force_device flag but it's not.

I am not sure where the problem is as I'm new to Theano, I appreciate your help.

Upvotes: 2

Views: 153

Answers (1)

Sonam Singh
Sonam Singh

Reputation: 26

Issue is floatX=float64. Use floatX=float32. GPU supports 32 bit only yet.

Upvotes: 1

Related Questions