【发布时间】：2018-07-01 07:50:06
【问题描述】：
我正在尝试运行下面的代码但报错：
NvvmSupportError：找不到 libNVVM。做conda install
cudatoolkit: library nvvm not found
我的开发环境是：Ubuntu 17.04、Spyder/Python3.5，我已经通过 conda（numba 和 cudatoolkit）安装。 Nvidia GPU（GTX 1070 和 GTX 1060）。
import numpy as np
from timeit import default_timer as timer
from numba import vectorize
@vectorize(["float32(float32, float32)"], target='cuda')
def VecADD(a,b):
return a+b
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 = VecADD(a,b)
print (timer() - start)
有人知道如何解决这个问题吗？
【问题讨论】：
-
听起来像 GPU 驱动程序或 CUDA 工具包没有正确安装。安装 CUDA 工具包后，是否正确设置了 PATH 和 LD_LIBRARY_PATH 环境变量？您是否验证了 CUDA 工具包安装？ /usr/local/cuda/nvvm/lib64 是否有任何 libnvvm 库？
-
我怀疑（如果我没记错的话），您不需要指定首先使用哪个 GPU 吗？
-
@stucash：不，你没有。发布的代码对我来说是开箱即用的
-
@talonmies 是的，我想我错了 :)
-
我做了几次测试，我相信这是一个PATH问题，但我仍然无法解决它。无论如何，我在 jupyter-notebook 上对其进行了测试，它运行良好，我现在就使用它。谢谢！
标签： python ubuntu cuda gpu numba