mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-13 02:12:27 +08:00
[NIXL][XPU] update install script of NIXL (#28778)
Signed-off-by: zhenwei-intel <zhenwei.liu@intel.com>
This commit is contained in:
parent
561253b37f
commit
d64429bb36
@ -14,6 +14,7 @@ RUN apt clean && apt-get update -y && \
|
|||||||
libxext6 \
|
libxext6 \
|
||||||
libgl1 \
|
libgl1 \
|
||||||
lsb-release \
|
lsb-release \
|
||||||
|
libaio-dev \
|
||||||
numactl \
|
numactl \
|
||||||
wget \
|
wget \
|
||||||
vim \
|
vim \
|
||||||
@ -68,8 +69,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
RUN python3 -m pip install -e tests/vllm_test_utils
|
RUN python3 -m pip install -e tests/vllm_test_utils
|
||||||
|
|
||||||
# install nixl from source code
|
# install nixl from source code
|
||||||
|
ENV NIXL_VERSION=0.7.0
|
||||||
RUN python3 /workspace/vllm/tools/install_nixl_from_source_ubuntu.py
|
RUN python3 /workspace/vllm/tools/install_nixl_from_source_ubuntu.py
|
||||||
ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/lib/python3.12/dist-packages/.nixl.mesonpy.libs/plugins/"
|
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/root/.cache/pip \
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
||||||
pip uninstall oneccl oneccl-devel -y
|
pip uninstall oneccl oneccl-devel -y
|
||||||
|
|||||||
@ -175,6 +175,7 @@ def build_and_install_prerequisites(args):
|
|||||||
build_env["LD_LIBRARY_PATH"] = (
|
build_env["LD_LIBRARY_PATH"] = (
|
||||||
f"{ucx_lib_path}:{ucx_plugin_path}:{existing_ld_path}".strip(":")
|
f"{ucx_lib_path}:{ucx_plugin_path}:{existing_ld_path}".strip(":")
|
||||||
)
|
)
|
||||||
|
build_env["LDFLAGS"] = "-Wl,-rpath,$ORIGIN"
|
||||||
print(f"--> Using LD_LIBRARY_PATH: {build_env['LD_LIBRARY_PATH']}", flush=True)
|
print(f"--> Using LD_LIBRARY_PATH: {build_env['LD_LIBRARY_PATH']}", flush=True)
|
||||||
|
|
||||||
temp_wheel_dir = os.path.join(ROOT_DIR, "temp_wheelhouse")
|
temp_wheel_dir = os.path.join(ROOT_DIR, "temp_wheelhouse")
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user