mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-01-23 21:04:31 +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 \
|
||||
libgl1 \
|
||||
lsb-release \
|
||||
libaio-dev \
|
||||
numactl \
|
||||
wget \
|
||||
vim \
|
||||
@ -68,8 +69,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
||||
RUN python3 -m pip install -e tests/vllm_test_utils
|
||||
|
||||
# install nixl from source code
|
||||
ENV NIXL_VERSION=0.7.0
|
||||
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 \
|
||||
pip uninstall oneccl oneccl-devel -y
|
||||
|
||||
@ -175,6 +175,7 @@ def build_and_install_prerequisites(args):
|
||||
build_env["LD_LIBRARY_PATH"] = (
|
||||
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)
|
||||
|
||||
temp_wheel_dir = os.path.join(ROOT_DIR, "temp_wheelhouse")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user