rapidsai-wheels-nightly / packages / ucx-py-cu12 0.43.0a2

0

Standard Python
Files
Labels
Badges

28074 total downloads
Last upload: 20 minutes and a few seconds ago

Installers

pip install

To install this package run one of the following:
pip install -i https://pypi.anaconda.org/rapidsai-wheels-nightly/simple ucx-py-cu12

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"

rapidsai-wheels-nightly / packages / ucx-py-cu12 0.43.0a2 0