cuQuantum Python:NVIDIA cuQuantum SDK 的 Python API

欢迎使用 cuQuantum Python 文档!

NVIDIA cuQuantum Python 提供 Python 绑定和高级面向对象模型,用于从 Python 访问 NVIDIA cuQuantum SDK 的全部功能。

可在以下位置获取https://github.com/NVIDIA/cuQuantum

主要功能

  • 为所有 cuQuantum C API 提供 1:1 Python 包装器

  • 为创建和操作张量网络提供高级、Pythonic 的对象

    • 可与 CPU (NumPy、PyTorch) 和 GPU (CuPy、PyTorch) 数组互操作

  • 为 Cirq 和 Qiskit 用户提供转换器,将量子电路对象映射到张量网络收缩

  • 为加速基于量子多体算符和密度矩阵(或状态向量)形式的模拟量子动力学求解器提供高级、Pythonic 的 API

  • 开源(在 BSD-3-Clause 许可下),遵循社区实践,易于访问

支持

  • 支持的 Python 版本: 3.10, 3.11, 3.12

  • 支持的安装方法conda installpip install 或从源代码构建

先决条件

  • CUDA 11 和 12

  • cuStateVec 1.7.0+

  • cuTensorNet 2.6.0+

  • cuDensityMat >=0.0.5, <0.1.0

  • NumPy v1.21+

  • CuPy v13.0+

  • 可选依赖项