2015-11-23 4 views
2

я пытаюсь установить PyCuda с 7.0 CUDA без проблем и запуска кода хорошо, но я не могу установить и запустить PyCuda после установки PyCuda с Installing PyCUDA on Ubuntu Linux попробуйте запустить простой код с PyCuda простого кодомустановить PyCuda в Ubuntu 14.04 с CUDA 7.0

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 

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

ответ

0

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

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 
# or somewhere else your cuda library is located 
Смежные вопросы