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