using GPU in theano operation

Skip to first unread message

Dominik Otto

May 2, 2019, 7:22:30 AM5/2/19
to theano-users
I would like to use multiple GPUs in my custom Theano operation as in
However, `theano.misc.pycuda_init` seems to be missing and initializing Pycuda after Theano produces memory issues unless I restrict them to there respective half.

What is the proper way to use the GPU in the `preform` method of my Theano operations?
Reply all
Reply to author
0 new messages