Tensorflow: Model trained(checkpoint files) on GPU

2019-09-26 08:30发布

问题:

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?

回答1:

Yes! It normally can!

The exception is with tf.device('gpu:0') statements. If you do not have them in your code you are good to go!

Good luck!