ucx-py-cu120.45.0
ucx-py-cu120.45.0
Published
Python Bindings for the Unified Communication X library (UCX)
pip install ucx-py-cu12
Package Downloads
Authors
Project URLs
Requires Python
>=3.10
Dependencies
- libucx-cu12
<1.19,>=1.15.0 - numpy
<3.0a0,>=1.23 - pynvml
<13.0.0a0,>=12.0.0 - cloudpickle
; extra == "test" - cudf-cu12
==25.8.*; extra == "test" - cupy-cuda12x
>=12.0.0; extra == "test" - distributed
; extra == "test" - numba
<0.62.0a0,>=0.59.1; extra == "test" - pytest-asyncio
; extra == "test" - pytest-rerunfailures
; extra == "test" - pytest
==7.*; extra == "test" - rapids-dask-dependency
==25.8.*; extra == "test"
[!WARNING] UCX-Py is deprecated and will be phased out soon, version 0.45 (RAPIDS 25.08) is expected to be the last release. Projects that rely on UCX-Py are advised to migrate to UCXX immediately.
Python Bindings for UCX
Installing
Users can either install with Conda or build from source.
Testing
To run ucx-py's tests, just use pytest:
pytest -v
TCP Support
In 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"