Tensorflow: 我可以将经过 gpu 训练的模型导出到仅使用 cpu 的机器上进行服务/评估吗?

创建于 2016-07-16  ·  3评论  ·  资料来源: tensorflow/tensorflow

我在 GPU 机器上训练了一个模型。
当我在仅使用 cpu 的机器上加载此类模型时,出现以下错误。

感谢您的任何建议!

awaiting tensorflower

最有用的评论

此外,在未来,StackOverflow 是提出这些问题的正确场所。

所有3条评论

@vrv你能看看这个吗? 我们现在有这样的支持吗? 谢谢。

是的,如果您真的想忽略图中的设备放置指令,您可以将 tf.ConfigProto(allow_soft_placement=True) 传递给 tf.Session。 我不太喜欢使用这个选项(我宁愿重写图形以明确地去除设备字段),但这应该可行。

此外,在未来,StackOverflow 是提出这些问题的正确场所。

此页面是否有帮助?
0 / 5 - 0 等级