Python 使用cuda 7.0在ubuntu 14.04中安装pycuda
我正在尝试用cuda 7.0安装pycuda没有问题,运行代码很好,但在安装pycuda后,我无法安装并运行pycuda,请尝试使用pycuda运行简单代码。简单代码为Python 使用cuda 7.0在ubuntu 14.04中安装pycuda,python,pycuda,Python,Pycuda,我正在尝试用cuda 7.0安装pycuda没有问题,运行代码很好,但在安装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
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
回溯(最近一次呼叫最后一次):
文件“/home/pupuol/Documents/educuda.py”,第3行,在
输入pycuda。自动初始化
文件“/usr/local/lib/python2.7/dist packages/pycuda-2015.1.3-py2.7-linux-x86_64.egg/pycuda/autoinit.py”,第2行,在
将pycuda.driver导入为cuda
文件“/usr/local/lib/python2.7/dist packages/pycuda-2015.1.3-py2.7-linux-x86_64.egg/pycuda/driver.py”,第5行,在
来自pycuda._驱动程序导入*#noqa
ImportError:libcurand.so.7.0:无法打开共享对象文件:没有此类文件或目录
请帮助我这似乎是一个库路径问题。 在运行python之前,请尝试此操作
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
# or somewhere else your cuda library is located