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
评论