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