'use a new way to init pycuda to work correctly with newer version of Theano and Pycuda.'

theano/misc/pycuda_example.py

@@ -22,13 +22,15 @@ from theano.sandbox.cuda import GpuElemwise, CudaNdarrayType
 from theano.sandbox.cuda.basic_ops import as_cuda_ndarray_variable, gpu_contiguous
 from theano.sandbox.cuda.opt import gpu_seqopt
 
+import pycuda_init
+if not pycuda_init.pycuda_available:
+    raise Exception("No pycuda available. You can't load pycuda_example.py")
+
+import pycuda
 from pycuda.elementwise import ElementwiseKernel
 from pycuda.compiler import SourceModule
-from pycuda.gpuarray import splay
 from pycuda.tools import VectorArg
 
-import pycuda.autoinit
-
 def theano_parse_c_arg(c_arg):
     c_arg = c_arg.replace('npy_float32','float')
     c_arg = c_arg.replace('npy_float64','double')

theano/misc/pycuda_init.py

+import os
+
+import theano
+
+def select_gpu_from_theano():
+    # Transfer the theano gpu binding to pycuda, for consistency
+    theano_to_pycuda_device_map = {"cpu": "0",
+                                   "gpu0": "0",
+                                   "gpu1": "1",
+                                   "gpu2": "2",
+                                   "gpu3": "3"}
+    os.environ["CUDA_DEVICE"] = theano_to_pycuda_device_map.get(theano.config.device, "0")
+
+select_gpu_from_theano()
+pycuda_available = False
+try:
+    import pycuda
+    import pycuda.autoinit
+    pycuda_available = True
+except ImportError:
+    pass

theano/misc/tests/test_pycuda.py

 import numpy
 
-try:
-    import pycuda
-except ImportError:
+import theano
+import theano.misc.pycuda_init
+
+if not theano.misc.pycuda_init.pycuda_available:
     from nose.plugins.skip import SkipTest
     raise SkipTest("Pycuda not installed. Skip test of theano op with pycuda code.")