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のみの2次元相関実装です。

私はCUDA Toolkit 7.5をインストールしていますが、TensorflowはGPUで完全に動作します。 何らかの理由でTheanoがCPUに落ちているので、force_deviceフラグのためにエラーが発生するはずですが、そうではありません。

私がTheanoに初めているときに問題がどこにあるのか分かりません。私はあなたの助けに感謝します。

Answer 1

問題はfloatX = float64です。 floatX = float32を使用します。 GPUはまだ32ビットしかサポートしていません。