驚呆了!這樣可以將Numpy加速700倍!
號外: 本號免費提供 CSDN 資源下載,需要的伙伴公眾號后臺回復(fù)【CSDN】
選自towardsdatascience,?作者:George Seif 本文轉(zhuǎn)自機(jī)器之心(nearhuman2014)

pip?install?cupy
i7–8700k CPU
1080 Ti GPU
32 GB of DDR4 3000MHz RAM
CUDA 9.0
import?numpy?as?np
import?cupy?as?cp
import?time
###?Numpy?and?CPU
s?=?time.time()
*x_cpu?=?np.ones((1000,1000,1000))*
e?=?time.time()
print(e?-?s)###?CuPy?and?GPU
s?=?time.time()
*x_gpu?=?cp.ones((1000,1000,1000))*
e?=?time.time()
print(e?-?s)
###?Numpy?and?CPU
s?=?time.time()
*x_cpu?*=?5*
e?=?time.time()
print(e?-?s)###?CuPy?and?GPU
s?=?time.time()
*x_gpu?*=?5*
e?=?time.time()
print(e?-?s)
數(shù)組乘以 5
數(shù)組本身相乘
數(shù)組添加到其自身
###?Numpy?and?CPU
s?=?time.time()
*x_cpu?*=?5
x_cpu?*=?x_cpu
x_cpu?+=?x_cpu*
e?=?time.time()
print(e?-?s)###?CuPy?and?GPU
s?=?time.time()
*x_gpu?*=?5
x_gpu?*=?x_gpu
x_gpu?+=?x_gpu*
e?=?time.time()
print(e?-?s)

推薦閱讀 關(guān)于 Python 3.9,那些你不知道的事 總結(jié)了 90 條寫 Python 程序的建議
關(guān)注「Python 知識大全」,做全棧開發(fā)工程師 歲月有你 惜惜相處
評論
圖片
表情
