2026-03-02 18:46:04 +08:00
|
|
|
FROM vllm_kunlun:custom_base_v0.0.3
|
2026-02-05 19:36:06 +08:00
|
|
|
|
|
|
|
|
WORKDIR /workspace
|
|
|
|
|
|
|
|
|
|
ARG PIP_INDEX_URL="https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
|
|
|
|
|
|
|
|
|
|
# Install vLLM
|
|
|
|
|
RUN uv pip install vllm==0.11.0 --no-build-isolation --no-deps -i ${PIP_INDEX_URL}
|
|
|
|
|
|
|
|
|
|
# Install vllm-kunlun and vxpu
|
|
|
|
|
COPY . /workspace/vLLM-Kunlun/
|
|
|
|
|
|
|
|
|
|
RUN cd /workspace/vLLM-Kunlun/ \
|
|
|
|
|
&& uv pip install -r requirements.txt -i ${PIP_INDEX_URL} \
|
|
|
|
|
&& python setup.py build_ext \
|
|
|
|
|
&& uv pip install -e . \
|
|
|
|
|
&& cd /workspace/vLLM-Kunlun/vllm_kunlun/csrc/vxpu_offload/ \
|
|
|
|
|
&& make install && make clean
|
|
|
|
|
|
|
|
|
|
CMD ["/bin/bash"]
|