如何在pip从源码安装时跳过构建wheel以避免大模型权重导致内存错误

你遇到的内存错误是因为pip默认安装源码包时会先构建二进制wheel包，构建过程需要将所有包内文件压缩为zip格式，超大的模型权重文件在压缩时一次性加载到内存触发了上限。

方案1：使用pip参数跳过wheel构建 #

直接在安装命令中添加--no-binary :all:参数强制走源码安装流程，完全跳过wheel构建和zip压缩步骤：

pip install --no-binary :all: ./mypackage --no-cache-dir

这个参数会让pip直接将包文件复制到目标安装目录，不会触发压缩逻辑。

方案2：直接调用setup.py安装 #

如果上述参数仍不生效，可以直接走旧版的setup.py安装流程，完全绕过pip的wheel构建逻辑：

cd mypackage/
python setup.py install

运行前请确认你已经激活了目标虚拟环境，避免安装到错误的Python环境中。

长期优化建议 #

如果需要频繁安装这个包，可以调整打包逻辑避免打包大文件：

• 将模型权重从package_data中移除，在setup.py中添加自定义安装钩子，安装时直接将权重文件夹复制到site-packages的对应路径，不参与打包流程
• 给包添加首次运行的权重初始化逻辑，运行时从本地固定路径加载权重，不需要将权重打进安装包内

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