CuPy 基于CUDA实现与NumPy兼容的多维数组的实现

CuPy是基于CUDA实现与NumPy兼容的多维数组的实现。 CuPy由核心多维数组类cupy.ndarray和许多函数组成。 它支持numpy.ndarray接口的子集。

安装：

(Binary Package for CUDA 8.0)
$ pip install cupy-cuda80

(Binary Package for CUDA 9.0)
$ pip install cupy-cuda90

(Binary Package for CUDA 9.1)
$ pip install cupy-cuda91

(Binary Package for CUDA 9.2)
$ pip install cupy-cuda92

(Binary Package for CUDA 10.0)
$ pip install cupy-cuda100

(Binary Package for CUDA 10.1)
$ pip install cupy-cuda101

(Source Package)
$ pip install cupy