Tensorflow:GPU上训练有素的模型(检查点文件)可以转换为CPU运行模型吗? [英] Tensorflow: Model trained(checkpoint files) on GPU can be converted to CPU running model?

查看:516
本文介绍了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屋!

查看全文
登录 关闭
扫码关注1秒登录
发送“验证码”获取 | 15天全站免登陆