为什么Python在Python3上加速计算点产品的速度比普通的NumPy慢?我使用的是accelerate版本2.3.1,安装了accelerate cudalib 2.0,Python3.5.2 Windows 10 64位。
import numpy as np
from accelerate.cuda.blas import dot as gpu_dot
import time
def numpydot():
start= time.time()
for i in range(100):
np.dot(np.arange(1000000, dtype=np.float64), np.arange(1000000, dtype=np.float64))
elapsedtime = time.time()-start
return elapsedtime
def acceleratedot():
start= time.time()
for i in range(100):
gpu_dot(np.arange(1000000, dtype=np.float64), np.arange(1000000, dtype=np.float64))
elapsedtime = time.time()-start
return elapsedtime
numpydot()
0.6446375846862793
acceleratedot()
1.33168363571167
我发现共享数组是用Numba创建的,Numba是一个单独的库。他们的网站上有文件。在
相关问题 更多 >
编程相关推荐