Python Bindings for the Unified Communication X library (UCX)
Project description
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"
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
ucx_py_cu11-0.37.0.tar.gz
(1.3 kB
view hashes)