Caution
UCX-Py has been discontinued, version 0.45 (RAPIDS 25.08) was its last release and it will receive no further updates, see the RAPIDS Support Notice 50 for details. Projects depending on UCX-Py are advised to migrate to UCXX immediately. UCXX documentation.
Users can either install with Conda or build from source.
To run ucx-py's tests, just use pytest:
pytest -vIn order to use TCP add tcp to UCX_TLS and set UCXPY_IFNAME to the network interface you want to use. Some setup examples:
# TCP using "eth0" and CUDA support
export UCX_TLS=tcp,cuda_copy,cuda_ipc
export UCXPY_IFNAME="eth0"
# InfiniBand using "ib0" and CUDA support
export UCX_TLS=rc,cuda_copy,cuda_ipc
export UCXPY_IFNAME="ib0"
# TCP using "eno0" and no CUDA support
export UCX_TLS=tcp
export UCXPY_IFNAME="eno0"