Windows 10下用PyInstaller部署TensorFlow模型遇模块缺失错误求助

我之前也碰到过一模一样的坑，在TensorFlow 1.7 + PyInstaller的组合下这个问题特别常见——主要是PyInstaller的自动依赖检测机制没覆盖到TensorFlow的部分内部私有模块，好在你的CUDA 9.0+cuDNN 7的版本和TF1.7是匹配的，不用在这方面调整。下面是几个亲测有效的解决办法：

• 手动指定隐藏导入模块
  PyInstaller经常漏检TensorFlow的内部私有模块，你可以在打包命令里直接添加隐藏导入参数，强制让它包含缺失的模块：

  pyinstaller --hidden-import=tensorflow.python._pywrap_tensorflow_internal your_script.py
  

  如果习惯用spec文件打包，就在spec文件的hiddenimports列表里加上这个模块：

  hiddenimports=['tensorflow.python._pywrap_tensorflow_internal']
  

  之后用pyinstaller your_script.spec执行打包即可。

• 复制缺失的Pyd/DLL文件到打包目录
  有时候即使加了隐藏导入，对应的二进制文件还是没被打包进去。找到你Python环境的Lib\site-packages\tensorflow\python目录，找到_pywrap_tensorflow_internal.pyd文件（这是Windows下的Python扩展模块文件），把它复制到打包生成的dist文件夹里的程序根目录下，再运行程序试试。

• 降级PyInstaller到兼容版本
  新版PyInstaller对TensorFlow 1.x的兼容性不太友好，试试切换到PyInstaller 3.3版本，这个版本对TF1.7的支持很稳定：

  pip uninstall pyinstaller -y && pip install pyinstaller==3.3
  

• 优先用目录模式打包
  如果你之前用的是--onefile单文件打包模式，改成--onedir目录模式试试——单文件模式在加载TensorFlow这类复杂依赖时容易出现路径问题，目录模式更容易排查和修复缺失的文件。

内容的提问来源于stack exchange，提问作者Prog