Skip to content

Instantly share code, notes, and snippets.

@brockpalen

brockpalen/theano-gpu

Created January 20, 2014 20:15
Show Gist options
  • Save brockpalen/8528166 to your computer and use it in GitHub Desktop.
Save brockpalen/8528166 to your computer and use it in GitHub Desktop.
Download ZIP
Theano on GPU's
Raw
theano-gpu
#Example for Theano 0.6 on CAEN-HPC FluxG
#load modules
module load python/2.7.5 cuda/5.5 theano/0.6
#use device=gpu to let our system autoselect the GPU to use
#use device=cpu to run your code on another system
THEANO_FLAGS='floatX=float32,device=gpu' python script.py
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment