"Anaconda加速的点积比普通的NumPy慢2倍"

2024-04-19 18:17:55 发布

您现在位置:Python中文网/ 问答频道 /正文

为什么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

Tags: importforgputimedefasnpstart