解决用gpu训练好的神经网络,用tensorflow-cpu跑报错问题

训练的时候当然用gpu,速度快呀。
我想用cpu版的tensorflow跑一下,结果报错,这个错误不太容易看懂。
解决用gpu训练好的神经网络,用tensorflow-cpu跑报错问题
大概意思是没找到一些节点。

后来发现原因,用gpu和cpu保存的pb模型不太一样,但是checkpoints文件是通用的。

使用tensorflow-cpu再把checkpoints文件重新转换一下pb文件就可以了。

完美解决!