跳转到主要内容

统一通信X库(UCX)的Python绑定

项目描述

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

UCX的Python绑定

安装

用户可以选择使用Conda安装或从源码构建

测试

要运行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"

项目详情


下载文件

下载适用于您的平台的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源代码分发

ucx_py_cu12-0.39.2.tar.gz (1.3 kB 查看哈希)

上传时间: 源代码

支持