nvidia-nvshmem-cu12

NVSHMEM creates a global address space that provides efficient and scalable communication for NVIDIA GPU clusters.

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

NVSHMEM is a parallel programming interface based on OpenSHMEM that provides efficient and scalable communication for NVIDIA GPU clusters. NVSHMEM creates a global address space for data that spans the memory of multiple GPUs and can be accessed with fine-grained GPU-initiated operations, CPU-initiated operations, and operations on CUDA streams.