python – 带有Tensorflow的Pyinstaller为_checkpoint_ops.so文

import os

tensorflow_location = '/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/tensorflow'

tensorflow_binaries = []

for dir_name,sub_dir_list,fileList in os.walk(tensorflow_location): 
  for file in fileList:
    if file.endswith(".so"):
      full_file = dir_name + '/' + file
      print(full_file)
      tensorflow_binaries.append((full_file,'.'))

确保还将二进制文件添加到Analysis

a = Analysis(...,binaries=tensorflow_binaries,...)

并确保.exe / .app版本中包含二进制文件.这是一个简单的例子：

pyz = PYZ(a.pure)
exe = EXE(pyz,a.scripts,a.binaries,a.zipfiles,a.datas,name='TestApp',debug=True,strip=None,upx=True,console=True,icon='Test.icns')
app = BUNDLE(exe,name='Test.app',icon='Test.icns',bundle_identifier='com.Test.TestApp',info_plist={
        'NSHighResolutionCapable': 'True'
            },)

您需要做的最后一件事是修改tensorflow load_library.py文件以指向.app运行时期间二进制文件的相对路径.

该文件通常位于此处(相对于python安装)：site-packages / tensorflow / python / framework / load_library.py

您需要添加resource_path函数并修改load_op_library函数以将file_name指向相对路径(并忽略按子文件夹查找二进制文件).

def resource_path(relative_path):
    """Due to pyinstaller changing root dir of project filenames need to be processed in order to open properly

    Parameters
    ----------
    relative_path : str
        String containing filename of project item (ex: models/bladder...h5)

    Returns
    -------
    path : str
        path relative to .exe on local computer
    """
    import sys
    import os

    try:
        # PyInstaller creates a temp folder and stores path in _MEIPASS
        base_path = sys._MEIPASS
    except Exception:
        base_path = os.path.abspath(".")

    path = os.path.join(base_path,relative_path)
    return path

def load_op_library(library_filename):
  """Loads a TensorFlow plugin,containing custom ops and kernels.

  Pass "library_filename" to a platform-specific mechanism for dynamically
  loading a library. The rules for determining the exact location of the
  library are platform-specific and are not documented here. When the
  library is loaded,ops and kernels registered in the library via the
  `REGISTER_*` macros are made available in the TensorFlow process. Note
  that ops with the same name as an existing op are rejected and not
  registered with the process.

  Args:
    library_filename: Path to the plugin.
      Relative or absolute filesystem path to a dynamic library file.

  Returns:
    A python module containing the Python wrappers for Ops defined in
    the plugin.

  Raises:
    RuntimeError: when unable to load the library or get the python wrappers.
  """ 
  # REMOVE AFTER PYINSTALLER USE
  library_filename = resource_path(library_filename.split('/')[-1])

这将使tensorflow通过pyinstaller应用程序的相对路径获取二进制文件(参见：https://pythonhosted.org/PyInstaller/runtime-information.html#run-time-information).

然后,Tensorflow将获取由.spec文件打包的二进制文件.

记得在Pyinstaller打包后取消注释load_library.py中的library_filename = resource_path(library_filename.split(‘/’)[ – 1])行(否则Tensorflow在你的本地Python环境中运行时将无法找到二进制文件！ )