diff --git a/Dockerfile b/Dockerfile index f2b35da..68ac115 100644 --- a/Dockerfile +++ b/Dockerfile @@ -72,12 +72,12 @@ RUN pip install . && \ python3 -m pip wheel . --no-deps -w dist/ -# # builder-extras 阶段,vllm 安装加上 WORKDIR(放在 flashinfer 后) -# WORKDIR /opt -# RUN git clone -b v0.4.2 https://github.com/vllm-project/vllm.git && \ -# cd vllm && \ -# python3 -m pip install ".[triton]" --no-build-isolation && \ -# python3 -m pip wheel . --no-deps -w /tmp/vllm_wheels +# builder-extras 阶段,vllm 安装(兼容 torch 2.7.1) +WORKDIR /opt +RUN git clone -b v0.9.1 https://github.com/vllm-project/vllm.git && \ + cd vllm && \ + python3 -m pip install . --no-build-isolation --no-deps && \ + python3 -m pip wheel . --no-deps -w /tmp/vllm_wheels # ── 编译你本地 sglang 源码并打 wheel ─────────────────────────────────────── @@ -91,7 +91,7 @@ RUN mkdir -p /wheels && \ cp /tmp/torch_dist/torch*.whl /wheels/ && \ cp /opt/vision/dist/torchvision-*.whl /wheels/ && \ cp /opt/flashinfer/dist/flashinfer_python-*.whl /wheels/ && \ - # cp /tmp/vllm_wheels/vllm-*.whl /wheels/ && \ + cp /tmp/vllm_wheels/vllm-*.whl /wheels/ && \ cp /tmp/sg_wheels/sglang-*.whl /wheels/ ###############################################################################