为什么Anaconda加速计算点产品比Python 3上的Plain Numpy慢?我正在使用ACCELERATE_CUDALIB 2.0安装加速版2.3.1,Python 3.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创建的。他们的网站上有文档。