CuPy基于CUDA實(shí)現(xiàn)與NumPy兼容的多維數(shù)組的實(shí)現(xiàn)

CuPy是基于CUDA實(shí)現(xiàn)與NumPy兼容的多維數(shù)組的實(shí)現(xiàn)。 CuPy由核心多維數(shù)組類(lèi)cupy.ndarray和許多函數(shù)組成。 它支持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