[XPU][Dockerfile] Update UCX install (#53817)
Signed-off-by: zhenwei-intel <zhenwei.liu@intel.com>
liuzhenwei committed
Aug 27, 2026 at 12:34 UTC
f25c580af15986f7ccdaaf7639c15896e04cf384
1 file changed
+8
-8
docker/Dockerfile.xpu
+8
-8
@@ -188,17 +188,17 @@ RUN --mount=type=cache,target=/root/.cache/uv \
188
git clone --depth 1 --branch "${UCX_VERSION}" https://github.com/openucx/ucx /tmp/ucx_source && \
189
cd /tmp/ucx_source && \
190
bash autogen.sh && \
191
- ./configure --prefix=/tmp/ucx_install --with-ze=yes --enable-examples --enable-mt && \
191
+ ./configure --prefix=/opt/ucx --with-ze=yes --enable-examples --enable-mt && \
192
make CFLAGS="-Wno-error=incompatible-pointer-types" -j"$(nproc)" && make install && \
193
git clone --depth 1 --branch "${NIXL_VERSION}" https://github.com/ai-dynamo/nixl /tmp/nixl_source && \
194
cd /tmp/nixl_source && \
195
uv pip install --upgrade meson pybind11 patchelf && \
196
uv pip install -r requirements.txt && \
197
- PKG_CONFIG_PATH=/tmp/ucx_install/lib/pkgconfig \
198
- LD_LIBRARY_PATH=/tmp/ucx_install/lib \
197
+ PKG_CONFIG_PATH=/opt/ucx/lib/pkgconfig \
198
+ LD_LIBRARY_PATH=/opt/ucx/lib \
199
python -m pip wheel --no-deps . -w /tmp/nixl_wheels/ && \
200
- find /tmp/ucx_install -type f \( -name '*.a' -o -name '*.la' \) -delete && \
201
- rm -rf /tmp/ucx_install/{include,share,etc,bin} /tmp/ucx_install/lib/cmake \
200
+ find /opt/ucx -type f \( -name '*.a' -o -name '*.la' \) -delete && \
201
+ rm -rf /opt/ucx/{include,share,etc} /opt/ucx/lib/cmake \
202
/tmp/ucx_source /tmp/nixl_source
203
204
FROM vllm-base AS vllm-openai
@@ -209,10 +209,10 @@ ARG NIXL_VERSION=v1.3.2
209
210
# Copy compiled UCX runtime libraries and the pre-built NIXL wheel.
211
# No compiler or autotools are installed in this stage.
212
-COPY --from=ucx-nixl-build /tmp/ucx_install /tmp/ucx_install
212
+COPY --from=ucx-nixl-build /opt/ucx /opt/ucx
213
COPY --from=ucx-nixl-build /tmp/nixl_wheels /tmp/nixl_wheels
214
215
-ENV LD_LIBRARY_PATH=/tmp/ucx_install/lib:${LD_LIBRARY_PATH}
215
+ENV LD_LIBRARY_PATH=/opt/ucx/lib:${LD_LIBRARY_PATH}
216
217
# Install RDMA runtime libraries (no build tools) and the pre-built NIXL wheel.
218
# Do not uninstall/reinstall large Python packages here to avoid extra layer
@@ -230,7 +230,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
230
librdmacm1t64 \
231
&& rm -rf /var/lib/apt/lists/* \
232
&& uv pip install --no-deps /tmp/nixl_wheels/nixl*.whl \
233
- && uv pip install nixl==${NIXL_VERSION} && uv pip uninstall nixl-cu13 \
233
+ && uv pip install --no-deps nixl==${NIXL_VERSION} \
234
&& rm -rf /tmp/nixl_wheels
235
236
RUN --mount=type=cache,target=/root/.cache/uv \