python - 无法将设备分配给节点

Question

我关注了this tutoriel将我自己训练的 tensorflow 模型导出到 C++，当我调用 freeze_graph 时出现错误

I tensorflow/core/common_runtime/gpu/gpu_device.cc:838] Creating TensorFlow device (/gpu:0) -> (device: 0, name: TITAN X (Pascal), pci bus id: 0000:03:00.0)
...
tensorflow.python.framework.errors.InvalidArgumentError: Cannot assign a device to node 'save/Const_1': Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.
Colocation Debug Info:
Colocation group had the following types and devices: 
Identity: CPU 
Const: CPU 
     [[Node: save/Const_1 = Const[dtype=DT_STRING, value=Tensor<type: string shape: [] values: model>, _device="/device:GPU:0"]()]]
Caused by op u'save/Const_1', defined at:
...

GPU:0 被 Tensorflow 检测到并可用，所以我不明白错误从何而来。

有什么想法吗？

Answer 1

该错误意味着 op save/Const_1 正在尝试放置在 GPU 上，并且该节点没有 GPU 实现。事实上，Const 节点仅适用于 CPU，并且作为 Graph 对象的一部分存储，因此不能放置在 GPU 上。一种解决方法是使用 allow_soft_placement=True 运行，或者打开 pbtxt 文件并手动删除该节点的 device 行