安装 nx-cugraph#

本指南介绍如何安装 nx-cugraph 并在您的工作流程中使用它。

系统要求#

nx-cugraph 需要以下条件

  • Volta 架构或更高版本的 NVIDIA GPU，具备计算能力 7.0+

  • CUDA 11.2、11.4、11.5、11.8、12.0、12.2、12.5 或 12.8

  • Python >= 3.10

  • NetworkX >= 3.2（推荐使用 3.4 或更高版本）

有关系统要求的更多详细信息，请参阅RAPIDS 系统要求文档。

安装包#

阅读RAPIDS 快速入门指南，了解有关安装所有 RAPIDS 库的更多信息。

nx-cugraph 可以使用 conda 或 pip 安装。它包含在 RAPIDS metapackage 中，也可以单独安装。

Conda#

每夜构建版

conda install -c rapidsai-nightly -c conda-forge -c nvidia nx-cugraph

稳定版

conda install -c rapidsai -c conda-forge -c nvidia nx-cugraph

pip#

每夜构建版

pip install nx-cugraph-cu11 --extra-index-url https://pypi.anaconda.org/rapidsai-wheels-nightly/simple

稳定版

pip install nx-cugraph-cu11 --extra-index-url https://pypi.nvidia.com

注意

  • 上述 pip install 示例适用于 CUDA 11。要安装适用于 CUDA 12 的版本，请将 -cu11 替换为 -cu12