CODEWITHSUNDEEP
pythonimportcudapycuda
Bolster
Bolster

Reputation: 7906

PyCuda: Can import module, then I can't... (PyCUDA Samples)

Example code:

import pycuda.autoinit
import pycuda.driver as drv
import numpy

from pycuda.compiler import SourceModule
mod = SourceModule("""
__global__ void multiply_them(float *dest, float *a, float *b)
{
  const int i = threadIdx.x;
  dest[i] = a[i] * b[i];
}
""")

multiply_them = mod.get_function("multiply_them")

a = numpy.random.randn(400).astype(numpy.float32)
b = numpy.random.randn(400).astype(numpy.float32)

dest = numpy.zeros_like(a)
multiply_them(
        drv.Out(dest), drv.In(a), drv.In(b),
        block=(400,1,1), grid=(1,1))

print dest-a*b

Results:

Traceback (most recent call last):
  File "test.py", line 12, in <module>
    """)
  File "build/bdist.linux-x86_64/egg/pycuda/compiler.py", line 238, in __init__
  File "build/bdist.linux-x86_64/egg/pycuda/compiler.py", line 223, in compile
  File "build/bdist.linux-x86_64/egg/pycuda/compiler.py", line 149, in _find_pycuda_include_path
ImportError: No module named pycuda

Sounds simple enough, so lets test this. 

Python 2.7.1 (r271:86832, Feb 17 2011, 14:13:40) 
[GCC 4.3.4] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import pycuda
>>> pycuda
<module 'pycuda' from '/home/abolster/lib/python2.7/site-packages/pycuda-0.94.2-py2.7-linux-x86_64.egg/pycuda/__init__.pyc'>
>>>

Ok, thats weird...

Long story short, even stepping through the file line by line into the python console, nothing goes wrong until the actual execution of the mod=SourceModule() line.

(Final Traceback, I promise)

/home/abolster/lib/python2.7/site-packages/pycuda-0.94.2-py2.7-linux-x86_64.egg/pycuda/compiler.pyc in _find_pycuda_include_path()
    147 def _find_pycuda_include_path():
    148     from imp import find_module
--> 149     file, pathname, descr = find_module("pycuda")
    150 
    151     # Who knew Python installation is so uniform and predictable?

ImportError: No module named pycuda

So it looks like pycuda is getting different include dirs than runtime python, which shouldn't happen (as i understand it)

Any ideas? (Sorry for the long question)

Talonmies borought up a point about nvcc not being found; unless python is getting its envars from somewhere I can't think of, there's no reason it shouldn't :

[bolster@dellgpu src]$ which nvcc
~/cuda/bin/nvcc

Upvotes: 1

Views: 6851

Answers (3)

harrism
harrism

Reputation: 27809

Changing to Python 2.6 and reinstalling relevant modules fixed the problem for the OP.

Upvotes: 3

talonmies
talonmies

Reputation: 72349

There is nothing wrong with the code you are trying to run - it should work. My guess is that nvcc cannot be found. Make sure that the path to the nvcc executable is set in your environment before you try using pycuda.compiler.

Upvotes: 1

fabmilo
fabmilo

Reputation: 48330

I think you did not install the CUDA toolkit from nvidia and added the

/usr/local/cuda/lib/

to

LD_LIBRARY_PATH

find the the .so of the pycuda module and give us the output of:

>lld pycuda.so

Upvotes: 0

Related Questions