Simple demo of how to use scikit-cuda with multiprocessing.

multiprocessing_skcuda_demo.py

#!/usr/bin/env python                                                                        

"""                                                                                          
Simple demo of how to use scikit-cuda with multiprocessing.                                  
"""

import atexit
import multiprocessing as mp

import numpy as np
import pycuda.driver as drv
import pycuda.gpuarray as gpuarray
import skcuda.linalg as linalg

def my_proc(n):
    drv.init()
    dev = drv.Device(n)
    ctx = dev.make_context()
    atexit.register(ctx.pop)
    linalg.init()

    # Generate different pseudorandom sequences for each GPU:                                
    np.random.seed(n)

    a_gpu = gpuarray.to_gpu(np.random.rand(2, 2))
    b_gpu = gpuarray.to_gpu(np.random.rand(2, 2))
    print 'GPU %s: ' % n
    print linalg.dot(a_gpu, b_gpu)

if __name__ == '__main__':
    num_gpus = 2
    for i in range(num_gpus):
        p = mp.Process(target=my_proc, args=(i, ))
        p.start()