From e37f7b016458edfdc413207644f96efa0b98cc94 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Tue, 26 Nov 2024 22:04:38 -0600 Subject: [PATCH] prefer system installs of UCX libraries when using RAPIDS wheels (#421) Contributes to https://github.com/rapidsai/build-planning/issues/118 Should only be merged if / after we merge https://github.com/rapidsai/ucx-wheels/pull/13. That PR switches `libucx` wheels' loading behavior, such that `libucx.load_library()` (which `ucx-py` and `libucxx` call at import time) prefers the `libuc{m,p,s}.so` provided by the `libucx` wheels. https://github.com/rapidsai/ucx-wheels/pull/13 introduces an environment variable to control that... this proposes setting that, to continue preferring system-installed UCX libraries in devcontainers. --------- Co-authored-by: Vyas Ramasubramani --- .devcontainer/rapids.Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.devcontainer/rapids.Dockerfile b/.devcontainer/rapids.Dockerfile index 74cfe238..0a85aef3 100644 --- a/.devcontainer/rapids.Dockerfile +++ b/.devcontainer/rapids.Dockerfile @@ -28,6 +28,8 @@ RUN apt update -y \ ENV DEFAULT_VIRTUAL_ENV=rapids +ENV RAPIDS_LIBUCX_PREFER_SYSTEM_LIBRARY=true + FROM ${BASE} as conda-base ENV DEFAULT_CONDA_ENV=rapids