用cuda7.0在ubuntu14.04中安装pycuda

Question

我正在尝试用cuda7.0安装pycuda没问题，运行代码不错，但我无法在安装pycuda后安装并运行pycuda Installing PyCUDA on Ubuntu Linux尝试用pycuda运行简单代码简单代码是

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule
import numpy

mod = SourceModule("""
___global___ void doublify(float *a)
{
   int idx= threadIdx.x + threadIdx.y*4;
   a[idx] *= 2;
}
""")
a = numpy.random.randn(4,4)
a = a.astype(numpy.float32)
a_gpu = cuda.mem_alloc(a.nbytes)
cuda.memcpy_htod(a_gpu,a)

func = mod.get_function("doublify")
func(a_gpu, block=(4,4,1))

a_doubled = numpy.empty_like(a)
cuda.memcpy_dtoh(a_doubled, a_gpu)
print a_doubled
print a

但代码没有运行并显示此错误

^{pr2}$

请帮帮我

Answer 1

这似乎是一个库路径问题。 在运行python之前尝试一下。在

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
# or somewhere else your cuda library is located