Tensorflow:GPU上训练有素的模型(检查点文件)可以转换为CPU运行模型吗? [英] Tensorflow: Model trained(checkpoint files) on GPU can be converted to CPU running model?
本文介绍了Tensorflow:GPU上训练有素的模型(检查点文件)可以转换为CPU运行模型吗?的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
使用GPU训练模型,并通过检查点文件保存结果。保存的检查点文件可以由cpu-tensorflow运行吗?如果没有,可以转换保存的检查点文件以便在cpu-tensorflow中运行模型吗?
A model is trained with GPU and result is saved by checkpoint file. The saved checkpoint file can be run by cpu-tensorflow? If not, can convert the saved checkpoint file so as to run model in cpu-tensorflow?
推荐答案
是!
使用tf.device(’gpu:0’)语句的是例外。如果您的代码中没有它们,那就太好了!
The exception is with tf.device('gpu:0')
statements. If you do not have them in your code you are good to go!
祝您好运!
这篇关于Tensorflow:GPU上训练有素的模型(检查点文件)可以转换为CPU运行模型吗?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文