Numpy.GPU面向 Numpy 的 GPU 加速庫(kù)

一、項(xiàng)目介紹

Numpy.GPU是一個(gè)面向Numpy的Gpu加速庫(kù)，基于Cuda。

注：您必須擁有一塊NVIDIA的GPU才能享受加速效果。

二、安裝教程

1.從源碼安裝

pip install git+https://gitee.com/killf/numpy.gpu.git

三、使用說(shuō)明

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)

注意：首次執(zhí)行時(shí)可能有點(diǎn)慢，這是由于cuda的動(dòng)態(tài)編譯造成的，之后的運(yùn)行就會(huì)非?？臁?/p>