2

モデルをトレーニングして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 を初めて使用するため、問題がどこにあるのかわかりません。ご協力いただきありがとうございます。

4

1 に答える 1

1

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

于 2016-08-24T12:18:10.607 に答える