经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 程序设计 » Python3 » 查看文章
Python3.11二进制AI项目程序打包为苹果Mac App(DMG)-应用程序pyinstaller制作流程(AppleSilicon)
来源:cnblogs  作者:刘悦的技术博客  时间:2024/8/26 9:17:26  对本文有异议

众所周知,苹果MacOs系统虽然贵为Unix内核系统,但由于系统不支持N卡,所以如果想在本地跑AI项目,还需要对相关的AI模块进行定制化操作,本次我们演示一下如何将基于Python3.11的AI项目程序打包为MacOS可以直接运行的DMG安装包,可以苹果系统中一键运行AI项目。

MacOs本地部署AI项目

首先确保本地已经安装好 arm 内核的Python3.11程序,可以在Python官网进行下载和安装:python.org

这里以快手团队著名的表情迁移项目 LivePortrait 为例子,首先克隆快手团队官方的项目:

  1. git clone https://github.com/KwaiVGI/LivePortrait.git

进入项目的目录:

  1. cd LivePortrait

安装基于Mac系统的相关依赖:

  1. # for macOS with Apple Silicon users
  2. pip install -r requirements_macOS.txt

随后修改app.py文件,在代码上方加入环境变量的设置:

  1. # coding: utf-8
  2. """
  3. The entrance of the gradio for human
  4. """
  5. import os
  6. os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"

PYTORCH_ENABLE_MPS_FALLBACK=1 这个环境变量用于 PyTorch 中的 MPS(Metal Performance Shaders)加速功能。

MPS 是苹果公司为 macOS 和 iOS 设备提供的图形处理单元 (GPU) 框架,可以加速机器学习模型的训练和推理。

PYTORCH_ENABLE_MPS_FALLBACK=1 表示启用 MPS 回退功能。当 PyTorch 检测到设备支持 MPS 时,它会优先使用 MPS 进行加速。如果 MPS 无法使用,它会回退到 CPU 上运行。
简而言之,设置这个环境变量可以帮助 PyTorch 在支持 MPS 的设备上利用 GPU 加速,并在不支持 MPS 的设备上正常运行。

如果不单独设置这个变量,启用推理的时候会报错。

随后,启动推理页面进行测试:

  1. python3 app.py

注意,由于xpose暂不支持mps推理,所以mac版本不支持动物表情驱动,只支持人物的表情驱动。

如果推理没有问题,那么可以开始进行打包操作了。

MacOs本地打包AI项目

首先,安装pyinstaller库:

  1. pip3 install -U pyinstaller

随后,创建 app.spec 项目配置文件:

  1. # -*- mode: python ; coding: utf-8 -*-
  2. import sys
  3. sys.setrecursionlimit(5000)
  4. from PyInstaller.utils.hooks import collect_data_files
  5. datas = []
  6. datas += collect_data_files('gradio_client')
  7. datas += collect_data_files('gradio')
  8. a = Analysis(
  9. ['app.py',
  10. ],
  11. pathex=['/Users/liuyue/Downloads/LivePortrait_For_Mac'],
  12. binaries=[],
  13. datas=datas,
  14. hiddenimports=[],
  15. hookspath=[],
  16. hooksconfig={},
  17. runtime_hooks=[],
  18. excludes=[],
  19. noarchive=False,
  20. optimize=0,
  21. module_collection_mode={ 'gradio': 'py'}
  22. )
  23. pyz = PYZ(a.pure)
  24. exe = EXE(
  25. pyz,
  26. a.scripts,
  27. [],
  28. exclude_binaries=True,
  29. name='LivePortrait',
  30. icon='AnyConv.com__paints_logo.icns',
  31. debug=False,
  32. bootloader_ignore_signals=False,
  33. strip=False,
  34. upx=True,
  35. console=True,
  36. disable_windowed_traceback=False,
  37. argv_emulation=False,
  38. target_arch=None,
  39. codesign_identity=None,
  40. entitlements_file=None,
  41. )
  42. a.datas += Tree('./pretrained_weights', prefix='pretrained_weights')
  43. coll = COLLECT(
  44. exe,
  45. a.binaries,
  46. a.datas,
  47. strip=False,
  48. upx=True,
  49. upx_exclude=[],
  50. name='LivePortrait',
  51. )

这里按照 pyinstaller 官方文档对项目的入口文件,依赖文件,三方目录等进行声明。

接着运行打包命令:

  1. pyinstaller webui.spec

程序返回:

  1. 98124 INFO: Rewriting the executable's macOS SDK version (13.1.0) to match the SDK version of the Python library (12.1.0) in order to avoid inconsistent behavior and potential UI issues in the frozen application.
  2. 98125 INFO: Re-signing the EXE
  3. 98243 INFO: Building EXE from EXE-00.toc completed successfully.
  4. 98244 INFO: checking Tree
  5. 98244 INFO: Building Tree because Tree-00.toc is non existent
  6. 98244 INFO: Building Tree Tree-00.toc
  7. 98265 INFO: checking COLLECT
  8. 98266 INFO: Building COLLECT because COLLECT-00.toc is non existent
  9. 98266 INFO: Building COLLECT COLLECT-00.toc
  10. 108930 INFO: Building COLLECT COLLECT-00.toc completed successfully.

代表打包成功,在项目的 dist 目录下会生成可执行程序:

双击 LivePortrait 图标进行测试即可。

至此,程序就打包好了。

MacOs本地构建DMG安装包

随后,运行磁盘工具,新建一个磁盘文件:

注意格式必须是 mac os 扩展(日志式),体积需要大于2G

接着把刚才打包好的项目文件拷贝到新建的磁盘中即可。

随后推出磁盘,点击映像-》转换,对磁盘文件进行压缩。

最后我们得到一个压缩好的DMG安装文件:

  1. ? mac ll
  2. total 5328720
  3. -rw-r--r--@ 1 liuyue staff 2.5G 8 20 19:49 LivePortrait(已转换).dmg

在别的Mac电脑中双击安装包打开运行即可。

至此我们就走完了整个MacOS的AI项目程序制作流程,最后,奉上打包好的程序文件,与众乡亲同飨:

  1. 新版LivePortrait整合包(苹果MacOsAppleSilicon)图片引擎 https://pan.quark.cn/s/53c24cd845b9

原文链接:https://www.cnblogs.com/v3ucn/p/18379269

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号