Theano 回退到 CPU答案

【问题标题】：Theano falls back to CPUTheano 回退到 CPU
【发布时间】：2016-08-24 09:01:49
【问题描述】：

我正在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 的新手，感谢您的帮助。

【问题讨论】：

【解决方案1】：

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

【讨论】：