2017-12-28 8 views
1

私はwith tf.device('/gpu:0')の中に構築する大きなモデルを用意しています。可能であればすべての作業をそこに配置したいと思います。ただし、1.4から1.2に更新した後、tensorflowはすべての操作をCPUに置くことを奨励しています(log_device_placementによる)。私は柔らかいデバイスの配置をオフにすることはできませんので、gpu上に配置することができないいくつかのオプス(形状の計算に関連し、AdamOptimizerに関連する)があるように見えます。テンソルフローに与えることができるいくつかの中間的なオプションがありますので、gpuカーネルを持たないもの以外のすべての関数をgpuに配置します。GPUにTensorflowオプションが設定されていません

答えて

1

実際にテンソルフローを実行していないようですが、ピップパッケージ(https://github.com/tensorflow/tensorflow/issues/12388)のバグのためです。 pip uninstall tensorflow && pip install tensorflow-gpuの後はすべてうまくいくようです。

関連する問題