モデルをトレーニングして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 実装から取られた CPU のみの 2d 相関実装であり、Torch でも使用されます。
CUDA Toolkit 7.5 をインストールしました。Tensorflow は GPU で完全に動作します。force_device
なんらかの理由で Theano が CPU にフォールバックしているため、フラグが原因でエラーが発生するはずですが、そうではありません。
私は Theano を初めて使用するため、問題がどこにあるのかわかりません。ご協力いただきありがとうございます。