Tensorflow：GPU上的训练模型（检查点文件）可以转换为CPU运行模型？

-3

模型使用GPU训练，结果由检查点文件保存。保存的检查点文件可以通过cpu-tensorflow运行？如果没有，可以转换已保存的检查点文件以在cpu-tensorflow中运行模型？Tensorflow：GPU上的训练模型（检查点文件）可以转换为CPU运行模型？

来源

2017-03-22 liu bob

可能的重复[可以在cpu上使用CPU进行推理的模型，反之亦然？]（http://stackoverflow.com/questions/40980035/can-a-model-trained-on-gpu-used- cpu-for-inference-and-反之亦然） – kaufmanu

是的！它通常可以！

例外是with tf.device('gpu:0')陈述。如果你的代码中没有他们，那么你很好！

祝你好运！

来源

2017-03-22 13:47:42 rmeertens

如果您的代码中有设备分配，可以使用remove_device重写规则将它们从保存的GraphDef .pb文件中删除：https：//github.com/ tensorflow/tensorflow /树/主/ tensorflow /工具/ graph_transforms /＃remove_device –

Tensorflow：GPU上的训练模型（检查点文件）可以转换为CPU运行模型？

回答

相关问题