-
Notifications
You must be signed in to change notification settings - Fork 4
Installing THEANO
After installing the latest NVIDIA driver for your system https://github.com/xdhub/cifar-ten/wiki/Installing-the-latest-NVIDIA-drivers, the final step of the Theano w/GPU install is to install Theano. First of all you want to install "OpenBLAS - Open Basic Linear Algebra Subprograms", which Theano needs to fit tightly with it to operate. Also, OpenBLAS is faster than other free BLAS options, thus predicating its first placement in this install. sudo apt-get install python-pip libopenblas-dev git
Then, since Theano is only available from PIP, the best way to get it is through sudo pip install Theano
. Once this has completed, it is a relatively simple process to test Theano itself with python -c "import theano; theano.test()"
. After this, it's a simple matter of running THEANO_FLAGS=floatX=float32,device=gpu python /usr/lib/python2.*/site-packages/theano/misc/check_blas.py
or some equivalent. Watch for .../local/...
directories and other pathing errors, but this is the code to run. Try running it with device=cpu
as well, and note the differences.