我尝试使用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/_cfi_backend. x86_64-linux-gnu. so“。这两个版本应该相等;检查你的安装。
也许是这个原因
$which pip
/usr/bin/pip
2条答案
按热度按时间nx7onnlm1#
您必须尝试分别删除所有的cffi包和安装的版本
1.5.2
。或者您可以通过更新旧版本来确保版本一致。
希望对你有帮助
fkvaft9z2#
更新pip到最新版本解决了我的问题。