大家好,又见面了,我是你们的朋友全栈君。...来看几个例子:
import numpy as np
import cupy as cp
import time
x=np.ones((1024,512,4,4))*1024.
y=np.ones((...这里之所以要弄个20次的平均,是因为,最开始的几次计算会比较慢!后面的计算速度才是稳定的,cpu和gpu都有一定这个特性,这个原因cpu和gpu是不同!...4,4,4,1))*512.3254
x=cp.ones((4,4,4,4))*1024.
y=cp.ones((4,4,4,1))*512.3254
GPU失去了优势,所以也不是所有计算都需要放到gpu上来加速的...有时候cpu算一算也是可以的!
cupy种几乎包含了numpy种通常有的很多function了!所以基本上再用的时候只要把‘np’ 换成‘cp’就好了!