Theano退回到CPU

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這是一個從caffe的cpp實現中獲取並且也被Torch使用的僅CPU的二維相關實現。

我安裝了CUDA Toolkit 7.5，Tensorflow在GPU上完美工作。 由於某種原因Theano正在回落到CPU，它應該會導致一個錯誤，由於force_device標誌，但它不是。

我不確定問題出在哪裏，因爲我是Theano的新手，我感謝您的幫助。

Answer 1

問題是floatX = float64。 使用floatX = float32。 GPU僅支持32位。