Tensorflow:GPU上的模型训练(检查点文件)可以转换为CPU运行模型吗?

时间:2017-03-22 12:22:52

标签: tensorflow gpu cpu checkpoint

使用GPU训练模型,并通过检查点文件保存结果。保存的检查点文件可以通过cpu-tensorflow运行吗?如果没有,可以转换保存的检查点文件,以便在cpu-tensorflow中运行模型吗?

1 个答案:

答案 0 :(得分:0)

是的!它通常可以!

例外是with tf.device('gpu:0')语句。如果你的代码中没有它们,你就可以去了!

祝你好运!

相关问题