CuPy是基于CUDA实现与NumPy兼容的多维数组的实现。CuPy由核心多维数组类cupy.ndarray和许多函数组成。它支持numpy.ndarray接口的子集。
安装:
(BinaryPackageforCUDA8.0)$pipinstallcupy-cuda80(BinaryPackageforCUDA9.0)$pipinstallcupy-cuda90(BinaryPackageforCUDA9.1)$pipinstallcupy-cuda91(BinaryPackageforCUDA9.2)$pipinstallcupy-cuda92(BinaryPackageforCUDA10.0)$pipinstallcupy-cuda100(BinaryPackageforCUDA10.1)$pipinstallcupy-cuda101(SourcePackage)$pipinstallcupy
评论