Prhub

#44366 [docker] Stop using extra-index-url for flashinfer-jit-cache

原始 PR 作者 khluu 合并时间 2026-06-03 09:58 文件变更 1 提交数 1 评论 0 代码增减 +1 / -1

执行摘要

Dockerfile 中 flashinfer 安装索引 URL 修正

flashinfer-jit-cache 当前在 PyPI 上被隔离(quarantined),使用 --extra-index-url 可能导致 pip 从 PyPI 拉取到其他版本,不符合预期。PR body 引用 jstawinsky 的提示,将该包锁定到官方索引。

建议合并此 PR 以修复构建环境的依赖稳定性。属于基础设施微调,无需深入精读。

讨论亮点

该 PR 无 review 讨论,仅由 robertgshaw2-redhat 快速批准。

实现拆解

  1. docker/Dockerfile 中,将 flashinfer-jit-cache 的 pip install 命令的参数 --extra-index-url 替换为 --index-url
  2. 同时更新索引 URL 格式不变,仍使用 CUDA 版本号动态拼接。
文件 模块 状态 重要度
docker/Dockerfile Docker modified 2.38

关键源码片段

docker/Dockerfile infrastructure

修改 flashinfer-jit-cache 的安装参数,是本次 PR 唯一变更文件

# patch 片段(未提供完整上下文,基于 diff hunk 还原)
RUN --mount=type=cache,target=/opt/uv/cache \
    ARG FLASHINFER_VERSION=0.6.12 \
    && uv pip install --system flashinfer-jit-cache==${FLASHINFER_VERSION} \
    # 改为 --index-url,确保仅从官方索引安装,避免 PyPI 污染
    --index-url https://flashinfer.ai/whl/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.')

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

风险极低。仅修改 pip 安装时的索引选择行为,确保 flashinfer-jit-cache 从官方源安装,避免依赖冲突。由于使用的是官方源,不会影响其他包安装。

影响范围仅限于 Docker 构建环境,确保 flashinfer-jit-cache 版本正确性,对用户无直接影响。

低风险

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论