print('Your GPU has %s GBs of free memory' % str(freeGPUMemInGBs))
testData = shared(np.random.random((5000, 256 * 256)).astype(T.config.
    floatX), borrow=True)
How to calculate GPU memory usage in Theano?