cuda_theano.py

#!/usr/bin/env python
__author__ = 'arenduchintala'
import theano
import theano.tensor as T
import numpy as np
a = T.vector('a')
M = T.matrix('M')
func = theano.function(inputs=[a,M], outputs=T.dot(M,a))
a_np = np.float32(np.random.rand(10,))
M_np = np.float32(np.random.rand(10,10))
func(a_np, M_np)

cuda_torch.py

#!/usr/bin/env python
__author__ = 'arenduchintala'
import torch
from torch import cuda
import sys
print cuda.is_available()
#device = int(sys.argv[1]) #get via `free-gpu` in the command line
cuda.set_device(0)
print cuda.current_device()
#make a random tensor and copy it to the gpu
tmp = torch.rand(5,5)
tmp.cuda()