【矩阵计算GPU加速】numpy 矩阵计算利用GPU加速,cupy包

【矩阵计算GPU加速】numpy 矩阵计算利用GPU加速,cupy包CuPy项目地址:https://cupy.chainer.org/ 这个项目本来是用来支持Chainer这个深度学习框架的,但是开发者把这个“GPU计算包”单独分出来了,方便了大家!!!来看几个例子:importnumpyasnpimportcupyascpimporttimex=np.ones((1024,512,4,4))*1024.y=np.one…

大家好,又见面了,我是你们的朋友全栈君。

CuPy 项目地址:https://cupy.chainer.org/ 

这个项目本来是用来支持Chainer这个深度学习框架的,但是开发者把这个“GPU 计算包”单独分出来了,方便了大家!!!

来看几个例子:

import numpy as np
import cupy as cp
import time

x=np.ones((1024,512,4,4))*1024.
y=np.ones((1024,512,4,1))*512.3254
time1=time.time()
for i in range(20):
    z=x*y
print('average time for 20 times cpu:',(time.time()-time1)/20.)

x=cp.ones((1024,512,4,4))*1024.
y=cp.ones((1024,512,4,1))*512.3254
time1=time.time()
for i in range(20):
    z=x*y
print('average time for 20 times gpu:',(time.time()-time1)/20.)

这里之所以要弄个20次的平均,是因为,最开始的几次计算会比较慢!后面的计算速度才是稳定的,cpu和gpu都有一定这个特性,这个原因cpu和gpu是不同!和“操作系统的本身算法、GPU工作方式”等有关系吧?

【矩阵计算GPU加速】numpy 矩阵计算利用GPU加速,cupy包

接下来,我们把矩阵规模减小

x=np.ones((4,4,4,4))*1024.
y=np.ones((4,4,4,1))*512.3254

x=cp.ones((4,4,4,4))*1024.
y=cp.ones((4,4,4,1))*512.3254

【矩阵计算GPU加速】numpy 矩阵计算利用GPU加速,cupy包

GPU失去了优势,所以也不是所有计算都需要放到gpu上来加速的!有时候cpu算一算也是可以的!

cupy种几乎包含了numpy种通常有的很多function了!所以基本上再用的时候只要把‘np’ 换成‘cp’就好了!很方便

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/153977.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

  • 炫酷的数据可视化_人力资源看板 数据可视化

    炫酷的数据可视化_人力资源看板 数据可视化12个超炫数据可视化工具今天我们带来一篇来自Adobe工程师RohitBoggarapu的文章。他在文章中介绍了一些适合网页开发者的数据可视化和绘图工具,让你不必再花大力气与枯燥的数据抗争。部分工具不要求写代码也可以使用!我们诠释数据的方式和数据本身之间存在着巨大的鸿沟。尤其是当我们唯一的选择是盯着表格中一列列不知所云的数字时。这可能是最无聊的一种格式了。没有哪个网页开发者会喜欢电子…

    2022年9月1日
    1
  • type-c是pd_type c pd 什么意思

    type-c是pd_type c pd 什么意思第一部分包含:一、二、三、四第二部分包含:五、六目录一、Type-C简介以及历史二、Type-CPort的DataRole、PowerRole三、Type-C的Data/PowerRole识别协商/AltMode四、如何进行数据链路的切换五、相关参数/名词/状态解释六、PD协议简介一、Type-C简介以及历史自1998年…

    2022年9月27日
    0
  • HSQL 学习笔记

    HSQL 学习笔记HSQL学习笔记1.    hsql 学习1.1.    学习目的本文档是针对hSQL 数据库方面的基础学习,为了使项目组成员能够达到使用hSQL 数据库的目的。1.2.    培训对象开发人员1.3.    常用词及符号说明常用词:hsql:一种免费的跨平台的数据库系统E:\hsqldb:表示是在dos 命令窗口下面1.4.    参考信息doc

    2022年9月22日
    0
  • Java后台开发知识一览

    Java后台开发知识一览Java后台开发知识一览1、后端WEB服务器:Weblogic、Tomcat、WebSphere、JBoss、Jetty核心框架:SpringFramework分布式服务框架Dubbo(感谢@浅浅浅丿忧伤指正)安全框架:ApacheShiro视图框架:SpringMVC服务端验证:Hibernate+Validator布局框架:SiteMesh工作流…

    2022年7月8日
    16
  • HDU 1754 I Hate It (段树单点更新)

    HDU 1754 I Hate It (段树单点更新)

    2022年1月7日
    40
  • 线程池与mq的简单结合使用

    线程池与mq的简单结合使用线程池与mq的简单结合使用

    2022年4月23日
    39

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号