-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Theano, CUDA and Ubuntu tutorial
I am running in Ubuntu 16.04 with CUDA 7.5.
The tutorial is here. However, it was not useful for my case: http://deeplearning.net/software/theano/install_ubuntu.html#install-ubuntu
The following instructions below did the job:
-
install theano:
sudo python3.5 -m pip install Theano
-
download and install the latest cuda: https://developer.nvidia.com/cuda-downloads
I had some issues with that, so I followed this idea (better option is to download the 1,1GB package as .run file):
You may also want to grab the proper NVidia driver, choose it form there:
System Settings > Software & Updates > Additional Drivers.
-
This should work, run it with:
THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python3.5 test.py
http://deeplearning.net/software/theano/tutorial/using_gpu.html -
Optionally, you can add cuDNN support from: https://developer.nvidia.com/cudnn