установить pycuda в ubuntu 14.04 с cuda 7.0

я пытаюсь установить pycuda с cuda 7.0 без проблем и запустить код хорошо, но я не могу установить и запустить pycuda после установки pycuda с помощью Установка PyCUDA в 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

но код не запускается и показывает эту ошибку

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

пожалуйста, помогите мне


person Pouya pourmohammad    schedule 23.11.2015    source источник


Ответы (1)


Похоже, это проблема пути к библиотеке. Попробуйте это, прежде чем запускать python.

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