• 授权协议:-
  • 开发厂商:-
  • 软件语言:Python
  • 更新日期:2018-11-09
Numpy.GPU

print注意:首次执行时可能有点慢,这是由于cuda的动态编译造成的,之后的运行就会非常快。

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的动态编译造成的,之后的运行就会非常快。

Numpy.GPU 面向 Numpy 的 GPU 加速库 评论内容