theanos use gpu

http://deeplearning.net/software/theano/install.html#gpu-linux

http://philipperemy.github.io/configure_gpu_for_deep_learning/

1) setup PATH and LD_LIBRARY_PATH

$ THEANO_FLAGS=cuda.root=/usr/local/cuda,mode=FAST_RUN,device=gpu,floatX=float32 \
 sudo -E python gpu_test.py

-E is used to pass environmental variables

for mnist testing, gpu is 50~60X faster than cpu

Advertisements

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / Change )

Twitter picture

You are commenting using your Twitter account. Log Out / Change )

Facebook photo

You are commenting using your Facebook account. Log Out / Change )

Google+ photo

You are commenting using your Google+ account. Log Out / Change )

Connecting to %s