Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
$ python gpu_theano.py
Using gpu device 0: GeForce GTX 1060 6GB (CNMeM is disabled, cuDNN 5110)
/home/kik/.local/lib/python2.7/site-packages/theano/sandbox/cuda/__init__.py:600: UserWarning: Your cuDNN version is more recent than the one Theano officially supports. If you see any problems, try updating Theano or downgrading cuDNN to version 5.
warnings.warn(warn)
[Elemwise{exp,no_inplace}(<TensorType(float32, vector)>), GpuFromHost(Elemwise{exp,no_inplace}.0)]
Looping 1000 times took 25.735937 seconds
Result is CudaNdarray([ 1.23178029 1.61879337 1.52278066 ..., 2.20771813 2.29967761
1.62323284])
Numpy result is [ 1.23178029 1.61879337 1.52278066 ..., 2.20771813 2.29967761
1.62323284]
Used the cpu
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.