CuPy是基于CUDA实现与NumPy兼容的多维数组的实现。CuPy由核心多维数组类cupy.darray和许多函数组成。它支持umpy.darray接口的子集。
安装:
(BiaryPackageforCUDA8.0)$pipistallcupy-cuda80(BiaryPackageforCUDA9.0)$pipistallcupy-cuda90(BiaryPackageforCUDA9.1)$pipistallcupy-cuda91(BiaryPackageforCUDA9.2)$pipistallcupy-cuda92(BiaryPackageforCUDA10.0)$pipistallcupy-cuda100(BiaryPackageforCUDA10.1)$pipistallcupy-cuda101(SourcePackage)$pipistallcupy






评论