使用pyinstaller对gradio和chromadb进行打包

顾德拉科2025-02-24 20:24

背景

python项目里包含了gradio和chromadb模块，使用pyinstaller后总有模块找不到，这里分享一个办法一招解决。

gradio在被打进exe后执行报错：

bash 复制代码

No such file or directory: gradio_client\types.json

这个主要原因是gradio和gradio_client相关模块没有被打进去。

还有一类报错：

bash 复制代码

[Errno 2] No such file or directory: gradio\blocks_events.pyc

这是因为在执行时默认去查找pyc文件，找不到报错。

chromadb在被打进exe后执行报错：

bash 复制代码

ModuleNotFoundError: No module named'chromadb.telemetry.product.posthog'

这主要是因为chromadb的相关子模块没有被包进去。

针对这两类问题，解决方案分两步：

1.输出.spec文件

bash 复制代码

pyinstaller --collect-data=gradio_client --collect-data=gradio --collect-data=chromadb -F rag_demo.py

2.在生成的rag_demo.spec文件中

Analysis部分添加

bash 复制代码

module_collection_mode={ 'gradio': 'py',}

因为我们希望 gradio 包的 .py 文件以某种特定方式被收集，比如直接将 .py 文件保留在打包后的文件结构中，而不是被编译成 .pyc 文件或者被压缩进 PYZ 文件里。