Numpy.GPU面向 Numpy 的 GPU 加速库
一、项目介绍
Numpy.GPU是一个面向Numpy的Gpu加速库,基于Cuda。
注:您必须拥有一块NVIDIA的GPU才能享受加速效果。
二、安装教程
1.从源码安装
pip install git+https://gitee.com/killf/numpy.gpu.git
三、使用说明
import numpy as np import numpy_gpu as gpu a = np.array([1, 2, 3]) b = np.array([4, 5, 6]) c1 = gpu.add(a, b) # [5. 7. 9.] c2 = gpu.dot(a, b) # 32 print(c1) print(c2)
注意:首次执行时可能有点慢,这是由于cuda的动态编译造成的,之后的运行就会非常快。
评论