跳到主要内容

cupy_cupy

cupy GitHub Repo stars​

CuPy 是一个兼容 NumPy/SciPy 的 Python GPU 加速计算库。它可作为现有 NumPy/SciPy 代码的无缝替代品，支持在 NVIDIA CUDA 和 AMD ROCm 平台上运行，并提供对底层 CUDA 特性的访问。