a1f142b7ad
There is a lot hack code for v0.11.0, which makes the code hard to
upgrade to newer vLLM version. Since v0.11.0 will release soon. Let's
drop v0.11.0 support first. Then we'll upgrade to v0.11.2 soon.
- vLLM version: v0.11.0
- vLLM main:
2918c1b49c
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
72 lines
3.2 KiB
Docker
72 lines
3.2 KiB
Docker
#
|
|
# Copyright (c) 2025 Huawei Technologies Co., Ltd. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# This file is a part of the vllm-ascend project.
|
|
#
|
|
|
|
FROM quay.io/ascend/cann:8.3.rc1-910b-ubuntu22.04-py3.11
|
|
|
|
ARG PIP_INDEX_URL="https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
|
|
ARG COMPILE_CUSTOM_KERNELS=1
|
|
ARG MOONCAKE_TAG="v0.3.7.post2"
|
|
|
|
# Define environments
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
ENV COMPILE_CUSTOM_KERNELS=${COMPILE_CUSTOM_KERNELS}
|
|
|
|
WORKDIR /workspace
|
|
|
|
COPY . /vllm-workspace/vllm-ascend/
|
|
|
|
# Install Mooncake dependencies
|
|
RUN apt-get update -y && \
|
|
apt-get install -y git vim wget net-tools gcc g++ cmake libnuma-dev && \
|
|
git clone --depth 1 --branch ${MOONCAKE_TAG} https://github.com/kvcache-ai/Mooncake /vllm-workspace/Mooncake && \
|
|
cp /vllm-workspace/vllm-ascend/tools/mooncake_installer.sh /vllm-workspace/Mooncake/ && \
|
|
cd /vllm-workspace/Mooncake && bash mooncake_installer.sh -y && \
|
|
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/Ascend/ascend-toolkit/latest/`uname -i`-linux/lib64 && \
|
|
mkdir -p build && cd build && cmake .. -DUSE_ASCEND_DIRECT=ON && \
|
|
make -j$(nproc) && make install && \
|
|
rm -fr /vllm-workspace/Mooncake/build && \
|
|
rm -rf /var/cache/apt/* && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN pip config set global.index-url ${PIP_INDEX_URL}
|
|
|
|
# Install vLLM
|
|
ARG VLLM_REPO=https://github.com/vllm-project/vllm.git
|
|
ARG VLLM_TAG=2918c1b49c88c29783c86f78d2c4221cb9622379
|
|
# Revert this change once VLLM_TAG is specified to branch or tag
|
|
# RUN git clone --depth 1 $VLLM_REPO --branch $VLLM_TAG /vllm-workspace/vllm
|
|
RUN git clone $VLLM_REPO /vllm-workspace/vllm && (cd /vllm-workspace/vllm && git checkout $VLLM_TAG)
|
|
# In x86, triton will be installed by vllm. But in Ascend, triton doesn't work correctly. we need to uninstall it.
|
|
RUN VLLM_TARGET_DEVICE="empty" python3 -m pip install -v -e /vllm-workspace/vllm/[audio] --extra-index https://download.pytorch.org/whl/cpu/ && \
|
|
python3 -m pip uninstall -y triton && \
|
|
python3 -m pip cache purge
|
|
|
|
# Install vllm-ascend
|
|
# Append `libascend_hal.so` path (devlib) to LD_LIBRARY_PATH
|
|
RUN export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi && \
|
|
source /usr/local/Ascend/ascend-toolkit/set_env.sh && \
|
|
source /usr/local/Ascend/nnal/atb/set_env.sh && \
|
|
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/Ascend/ascend-toolkit/latest/`uname -i`-linux/devlib && \
|
|
python3 -m pip install -v -e /vllm-workspace/vllm-ascend/ --extra-index https://download.pytorch.org/whl/cpu/ && \
|
|
python3 -m pip cache purge
|
|
|
|
# Install modelscope (for fast download) and ray (for multinode)
|
|
RUN python3 -m pip install modelscope 'ray>=2.47.1,<=2.48.0' 'protobuf>3.20.0' && \
|
|
python3 -m pip cache purge
|
|
|
|
CMD ["/bin/bash"]
|