【发布时间】:2020-02-21 11:03:24
【问题描述】:
我尝试使用 CUDA 运行代码,出现此错误,系统似乎有问题
完整代码:我知道了 CUDACast #10a - Your First CUDA Python Program 和 no module named numbapro
import numpy as np
from timeit import default_timer as timer
from numba import vectorize
@vectorize(["float32(float32, float32)"], target='cuda')
def VectorAdd(a, b):
return a + b
def main():
N = 32000000
A = np.ones(N, dtype=np.float32)
B = np.ones(N, dtype=np.float32)
C = np.zeros(N, dtype=np.float32)
start = timer()
C = VectorAdd(A, B)
vectoradd_timer = timer() - start
print("C[:5] = " + str(C[:5]))
print("C[-5:] = " + str(C[-5:]))
print("VectorAdd took %f seconds" % vectoradd_timer)
if __name__ == '__main__':
main()
输出:
例外:版本不匹配:这是“cffi”包版本 1.13.1,位于'/usr/local/lib/python2.7/dist-packages/cffi/api.pyc'。当我们导入 顶级 '_cffi_backend' 扩展模块,我们得到版本 1.5.2, 位于 '/usr/lib/python2.7/dist-packages/_cffi_backend.x86_64-linux-gnu.so'。 两个版本应该相等;检查您的安装。
可能是这个原因:
$which pip
/usr/bin/pip
【问题讨论】:
-
与cuda无关。只需检查它抱怨的两个软件包的版本