问一个在gpu上训练的模型可以用在cpu上进行推理，反之亦然？
EN

Stack Overflow用户

提问于 2016-12-05 17:34:06

回答 1查看 6.9K关注 0票数 16

我想知道在GPU上训练的模型是否可以用来与cpu一起运行推理？(反之亦然)感谢你！

回答已采纳

发布于 2016-12-05 17:41:30

您可以这样做，只要您的模型没有明确的设备分配。IE，如果您的模型有像with tf.device('gpu:0')这样的块，当您在没有GPU的模型上运行它时，它会抱怨。

在这种情况下，您必须确保导入的模型没有显式的设备分配，但在图表中使用图表参数

票数 17

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/40980035

复制

相似问题

问一个在gpu上训练的模型可以用在cpu上进行推理，反之亦然？EN