All Projects → rapidsai → ucx-py

rapidsai / ucx-py

Licence: BSD-3-Clause license
Python bindings for UCX

Programming Languages

python
139335 projects - #7 most used programming language
cython
566 projects
shell
77523 projects

https://ucx-py.readthedocs.io/en/latest/

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"
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].