使用 cuda 7.0 在 ubuntu 14.04 中安装 pycuda

Question

我正在尝试使用 cuda 7.0 安装 pycuda 没问题并且运行代码很好，但是在使用 Installing PyCUDA on Ubuntu Linux 安装 pycuda 后我无法安装和运行 pycuda 尝试使用 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

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

Traceback (most recent call last):
File "/home/pupuol/Documents/educuda.py", line 3, in <module>
import pycuda . autoinit
File "/usr/local/lib/python2.7/dist-packages/pycuda-2015.1.3-py2.7-    linux-x86_64.egg/pycuda/autoinit.py", line 2, in <module>
import pycuda.driver as cuda
File "/usr/local/lib/python2.7/dist-packages/pycuda-2015.1.3-py2.7-    linux-x86_64.egg/pycuda/driver.py", line 5, in <module>
from pycuda._driver import *  # noqa
ImportError: libcurand.so.7.0: cannot open shared object file: No such    file or directory

请帮帮我

Answer 1

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

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