1 Star 0 Fork 2

yutiansut / Numpy.GPU

forked from killf / Numpy.GPU 
加入 Gitee
与超过 600 万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README.md

一、项目介绍

Numpy.GPU是一个面向Numpy的Gpu加速库,基于Cuda。

注:您必须拥有一块NVIDIA的显卡(支持cuda)才能享受加速效果。

二、安装教程

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

仓库评论 ( 0 )

你可以在登录后,发表评论

简介

Numpy.GPU是一个面向Numpy的Gpu加速库,基于Cuda。 展开 收起
Python
Apache-2.0
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
Python
1
https://gitee.com/yutiansut/numpy.gpu.git
git@gitee.com:yutiansut/numpy.gpu.git
yutiansut
numpy.gpu
Numpy.GPU
master

搜索帮助