Help us improve your experience.

Let us know what you think.

Do you have time for a two-minute survey?

 
 

附录 A – 如何使用自动配置的 IPv6 地址运行 NCCL 测试

要使用静态或通过 SLAAC 自动分配的全局 IPv6 地址运行模型或 NCCL 测试,必须调整 NCCL_IB_GID_INDEX 变量的值。NCCL_IB_GID_INDEX 变量定义 RoCE (RDMA) 通信使用的全局 ID 索引。

默认值为 -1,这意味着 NCCL 将根据 InfiniBand 设备的活动链路层自动选择正确的 GID 索引。如果链路层是以太网 (RoCE),NCCL 将使用返回支持 RoCE v2 的 GID(通常为 GID 索引 3,具体取决于驱动程序/固件)的 GID 索引。

有关更多详细信息,您可以查看 Nvidia 的 环境变量 — NCCL 2.27.3 文档

要查找所需地址的 GID,请使用以下命令:

要查找 mellanox 接口名称,您可以使用以下脚本:

示例:

确定 GID 后,您可以运行 NCCL 测试,如示例所示:

以下脚本提供 Mellanox 接口名称、NIC编号和用户分配的接口名称(例如 gpu0_eth)之间的映射信息。它还提供接口与 GPU 之间的映射信息。

示例: