Tensorflow操作不会放在GPU上

Question

我有一个大型模型，我在with tf.device('/gpu:0')内构建，因为我希望所有操作都放在那里，如果可能的话。但是，在更新到1.4（从1.2）之后，tensorflow坚持将所有操作放在cpu上（根据log_device_placement）。我无法关闭软设备放置，因为似乎有些操作（与计算形状相关，以及AdamOptimizer）似乎无法放置在gpu上。是否有一些中间选项，我可以给tensorflow，以便它放置所有操作，除了没有gpu内核在gpu上？

Answer 1

由于pip打包中的一些错误（https://github.com/tensorflow/tensorflow/issues/12388），我似乎并没有真正运行tensorflow-gpu。在pip uninstall tensorflow && pip install tensorflow-gpu之后，一切似乎都运转正常。