Question

我正在Theano 0.9和Lasagne 0.1训练模型，并希望在GPU上运行它。我已将THEANO_FLAGS设置如下：

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

Theano打印它是使用GPU

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

但是，我注意到它没有，分析显示它使用CorrMM操作，这是根据文档

CorrMM这是一个仅限CPU的2d关联实现，取自caffe的cpp实现，也被Torch使用。

我安装了CUDA Toolkit 7.5，Tensorflow在GPU上完美运行。由于某种原因，Theano正在回归CPU，它应该会因force_device标志而导致错误，但不是。

由于我是Theano的新手，我不确定问题出在哪里，感谢您的帮助。

Answer 1

问题是floatX = float64。使用floatX = float32。 GPU仅支持32位。