Welcome to PyCUDA’s documentation!¶
PyCUDA gives you easy, Pythonic access to Nvidia’s CUDA parallel computation API. Several wrappers of the CUDA API already exist–so why the need for PyCUDA?
Object cleanup tied to lifetime of objects. This idiom, often called RAII in C++, makes it much easier to write correct, leak- and crash-free code. PyCUDA knows about dependencies, too, so (for example) it won’t detach from a context before all memory allocated in it is also freed.
Convenience. Abstractions like
pycuda.compiler.SourceModule
andpycuda.gpuarray.GPUArray
make CUDA programming even more convenient than with Nvidia’s C-based runtime.Completeness. PyCUDA puts the full power of CUDA’s driver API at your disposal, if you wish.
Automatic Error Checking. All CUDA errors are automatically translated into Python exceptions.
Speed. PyCUDA’s base layer is written in C++, so all the niceties above are virtually free.
Helpful Documentation. You’re looking at it. ;)
Here’s an example, to give you an impression:
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)
(This example is examples/hello_gpu.py
in the PyCUDA
source distribution.)
On the surface, this program will print a screenful of zeros. Behind the scenes, a lot more interesting stuff is going on:
PyCUDA has compiled the CUDA source code and uploaded it to the card.
Note
This code doesn’t have to be a constant–you can easily have Python generate the code you want to compile. See Metaprogramming.
PyCUDA’s numpy interaction code has automatically allocated space on the device, copied the numpy arrays a and b over, launched a 400x1x1 single-block grid, and copied dest back.
Note that you can just as well keep your data on the card between kernel invocations–no need to copy data all the time.
See how there’s no cleanup code in the example? That’s not because we were lazy and just skipped it. It simply isn’t needed. PyCUDA will automatically infer what cleanup is necessary and do it for you.
Curious? Let’s get started.
Contents¶
Note that this guide will not explain CUDA programming and technology. Please refer to Nvidia’s programming documentation for that.
PyCUDA also has its own web site, where you can find updates, new versions, documentation, and support.