2

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

4

1 回答 1

1

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

于 2016-08-24T12:18:10.607 回答