从源代码进行 pip 安装，无需构建轮子

Question

我有一个 Python 包，其中包含大型 PyTorch 模型检查点。我尝试将那些包含在我的setup.pyas中

package_data = {'mypackage': ['model_weights/*', 'model_weights/sequential_models*']},

现在的问题是，每当我尝试通过源安装时，pip install mypackage/ --no-cache-dir我都会得到一个MemoryError. 我尝试调试并--verbose意识到这发生在

creating '/tmp/pip-wheel-bs29bp6a/tmpp0itbxn1/mypackage-1.0-py3-none-any.whl' and adding 'build/bdist.linux-x86_64/wheel' to it
adding 'mypackage/model_weights/distilled_model.pt'
adding 'mypackage-1.0.dist-info/RECORD'
Traceback (most recent call last):
...
File "/zhome/1d/8/153438/miniconda3/envs/testenv/lib/python3.9/zipfile.py", line 1127, in write
      data = self._compressor.compress(data)
MemoryError
Building wheel for mypackage (PEP 517) ... error
ERROR: Failed building wheel for mypackage

我真的只想安装将文件复制到model_weights/安装目录中。将它们纳入车轮似乎是不可能的。

有没有办法在运行时抑制这一步pip install？该包只会作为源分发，而不会在 PyPI 上分发，因为model_weights文件太大了。

Answer 1

你可以运行

$ pip install mypackage/ --no-cache-dir --no-binary=mypackage

跳过轮子构建（假设mypackage实际上是您的发行版名称 - 这是您传递给name函数的名称setup()）。