This commit is contained in:
parent
79d4ccae04
commit
5242a5574f
14
Dockerfile
14
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/
|
||||
|
||||
###############################################################################
|
||||
|
|
|
|||
Loading…
Reference in New Issue