nvidia-nccl-cu12

NVIDIA Collective Communication Library (NCCL) Runtime

31 个版本 Python >=3
Nvidia CUDA Installer Team <compute_installer@nvidia.com>
安装
pip install nvidia-nccl-cu12
poetry add nvidia-nccl-cu12
pipenv install nvidia-nccl-cu12
conda install nvidia-nccl-cu12
描述

NCCL (pronounced "Nickel") is a stand-alone library of standard collective communication routines for GPUs, implementing all-reduce, all-gather, reduce, broadcast, and reduce-scatter. It has been optimized to achieve high bandwidth on any platform using PCIe, NVLink, NVswitch, as well as networking using InfiniBand Verbs or TCP/IP sockets.