I want to learning deep learning, so config cuda is a essential step. luckily it is very easy in Ubuntu
install theano+cuda in ubuntu
1. install theano
a) sudo apt-get install python-numpy python-scipy python-dev python-pip python-nose g++ libopenblas-dev git
b) sudo pip install Theano
2. install cuda
a) sudo apt install nvidia-331
If it is successful, we can test it with
$dkms status
If you see the response like nvidia-331, 331.38, 3.13.0-24-generic, x86_64: installed
, congratulations.
b) sudo apt install nvidia-cuda-toolkit
c) Reboot your machine
d)test with
import theano theano.test()
If you see something like Using gpu device 0: GeForce GT 740M,OK
e) config
sudo vim ~/.theanorc
[global] floatX = float32 device = gpu
OK We can enjoy the gpu.
reference
时间: 2024-10-21 13:21:49