统一通信X库(UCX)的Python绑定
项目描述
UCX的Python绑定
安装
测试
要运行ucx-py的测试,只需使用pytest
pytest -v
TCP支持
为了使用TCP,将tcp
添加到UCX_TLS
并将UCXPY_IFNAME
设置为要使用的网络接口。一些设置示例
# 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"