Theano fällt zurück auf CPU

Question

Ich trainiere ein Modell in Theano 0.9 und Lasagne 0.1 und möchte es auf GPU laufen. Ich habe festgelegt THEANO_FLAGS wie folgt:

THEANO_FLAGS=device=gpu0,force_device=True,floatX=float64 

Theano druckt es mit GPU

Using gpu device 0: GeForce GTX 980 Ti (CNMeM is disabled, cuDNN 4007) 

aber ich bemerkte es nicht, Profilieren zeigt, dass es CorrMM Betrieb ist mit der auf die Dokumentation wird nach

CorrMM Dies ist eine Nur-CPU-2d-Korrelationsimplementierung, die aus caffes cpp-Implementierung stammt und auch von Torch verwendet wird.

Ich habe CUDA Toolkit 7.5 installiert, Tensorflow funktioniert perfekt auf GPU. Aus irgendeinem Grund Theano fällt zurück in die CPU, es soll einen Fehler aufgrund force_device Flag verursachen, aber es ist nicht.

Ich bin mir nicht sicher, wo das Problem ist, wie ich neu bei Theano bin, schätze ich Ihre Hilfe.

Answer 1

Problem ist floatX = float64. Verwenden Sie floatX = float32. GPU unterstützt nur 32 Bit.