TensorFlow 2.5.0导入tensorflow.tools.graph_transforms报ModuleNotFoundError问题问询
报错原因
tensorflow.tools.graph_transforms是TensorFlow 1.x版本的内置模块,在TensorFlow 2.0及之后的正式版本中已经从官方安装包中移除,所以你在2.5.0版本下导入会触发ModuleNotFoundError。
可选择的解决方案
- 方案一:降级到TensorFlow 1.x版本
该方案适合原有代码重度依赖TensorFlow 1.x生态的场景,执行以下命令安装带该模块的版本:pip install tensorflow==1.15.5
注意:TensorFlow 1.15系列最高兼容Python 3.7,你当前使用的Python 3.9无法适配,需要先降级Python版本到3.7及以下才能使用该方案。 - 方案二:替换为TensorFlow 2.x原生等价功能
如果你使用TransformGraph是为了做模型图优化、量化、裁剪类操作,可以直接使用TensorFlow 2.x内置的tf.lite.TFLiteConverter、tf.graph_util系列接口实现相同需求,无需额外安装依赖。 - 方案三:单独安装适配2.x的graph_transforms包
不想降级环境的话可以直接安装第三方适配的独立包,执行命令:pip install tensorflow-graph-transforms
安装完成后把原有导入代码修改为以下内容即可正常使用:from tensorflow_graph_transforms import TransformGraph
该包兼容TensorFlow 2.5+和Python 3.9版本,适配Windows 10环境。
内容的提问来源于stack exchange,提问作者Amit Kumar Datta
相关产品推荐
相关产品推荐

