TensorFlow 操作未放置在 GPU 上

Question

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

Answer 1

看起来我实际上并没有运行 tensorflow-gpu，因为它们的 pip 包装中存在一些错误 (https://github.com/tensorflow/tensorflow/issues/12388)。在pip uninstall tensorflow && pip install tensorflow-gpu 之后，一切似乎都正常了。