TFグラフが実際に GPU で実行されていることを確認する簡単な方法を探しています。
PS。cuDNNまた、ライブラリが使用されていることを確認するとよいでしょう。
13416 次
3 に答える
0
関連するTensorFlow アップストリームの問題があります。基本的に、Python API はそのような情報をまだ公開していないと言っています。
ただし、C++ API はそうです。例えば、ありますtensorflow::KernelsRegisteredForOp()。私はその周りに小さな Python ラッパーを書き、supported_devices_for_op ここ(このコミット内) に実装しました。
于 2017-12-01T12:49:38.013 に答える