Theano 回到 CPU
Theano falls back to CPU
我正在 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 This is a CPU-only 2d correlation implementation taken from caffe’s cpp implementation and also used by Torch.
我安装了 CUDA Toolkit 7.5，Tensorflow 在 GPU 上运行完美。
出于某种原因，Theano 正在回落到 CPU，它应该会由于 force_device 标志而导致错误，但事实并非如此。
我不确定问题出在哪里，因为我是 Theano 的新手，感谢您的帮助。
问题是 floatX=float64。
使用 floatX=float32。 GPU 目前仅支持 32 位。
我正在 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 This is a CPU-only 2d correlation implementation taken from caffe’s cpp implementation and also used by Torch.
我安装了 CUDA Toolkit 7.5，Tensorflow 在 GPU 上运行完美。
出于某种原因，Theano 正在回落到 CPU，它应该会由于 force_device 标志而导致错误，但事实并非如此。
我不确定问题出在哪里，因为我是 Theano 的新手，感谢您的帮助。
问题是 floatX=float64。 使用 floatX=float32。 GPU 目前仅支持 32 位。