EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[Bugfix] Change vLLM install order & Add A2 support (#9232)

Browse Source
This commit is contained in:
Even Zhou
2025-08-17 13:36:14 +08:00
committed by GitHub
parent 1df84ff414
commit fda762a27d
3 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
docker/Dockerfile.npu
View File

@@ -51,22 +51,21 @@ ENV LC_ALL=en_US.UTF-8
# Install dependencies
# TODO: install from pypi released memfabric
# TODO: install from pypi released triton-ascend
RUN pip install $MEMFABRIC_URL --no-cache-dir \
&& pip install torch==$PYTORCH_VERSION torchvision==$TORCHVISION_VERSION --index-url https://download.pytorch.org/whl/cpu --no-cache-dir \
&& wget ${PTA_URL} && pip install "./torch_npu-2.6.0.post1-cp311-cp311-manylinux_2_28_aarch64.whl" --no-cache-dir \
&& pip install ${TRITON_ASCEND_URL} --no-cache-dir \
&& python3 -m pip install --no-cache-dir numpy==1.26.4 pybind11
RUN pip install $MEMFABRIC_URL --no-cache-dir
# Install vLLM
RUN git clone --depth 1 https://github.com/vllm-project/vllm.git --branch $VLLM_TAG && \
cd vllm && VLLM_TARGET_DEVICE="empty" pip install -v . --no-cache-dir && \
cd .. && rm -rf vllm
(cd vllm && VLLM_TARGET_DEVICE="empty" pip install -v . --no-cache-dir) && rm -rf vllm
# TODO: install from pypi released triton-ascend
RUN pip install torch==$PYTORCH_VERSION torchvision==$TORCHVISION_VERSION --index-url https://download.pytorch.org/whl/cpu --no-cache-dir \
&& wget ${PTA_URL} && pip install "./torch_npu-2.6.0.post1-cp311-cp311-manylinux_2_28_aarch64.whl" --no-cache-dir \
&& python3 -m pip install --no-cache-dir attrs==24.2.0 numpy==1.26.4 scipy==1.13.1 decorator==5.1.1 psutil==6.0.0 pytest==8.3.2 pytest-xdist==3.6.1 pyyaml pybind11 \
&& pip install ${TRITON_ASCEND_URL} --no-cache-dir
# Install SGLang
RUN git clone https://github.com/sgl-project/sglang --branch $SGLANG_TAG && \
cd ./sglang/python && pip install .[srt_npu] --no-cache-dir && \
cd .. && rm -rf ./sglang
(cd sglang/python && pip install -v .[srt_npu] --no-cache-dir) && rm -rf sglang
# Install Deep-ep
RUN git clone --branch $SGLANG_KERNEL_NPU_TAG https://github.com/sgl-project/sgl-kernel-npu.git \