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

    技术2025-03-14  23

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

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

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

    完美解决!

    Processed: 0.010, SQL: 9