ubuntu14.04安装theano
基本上按照官网来就行:
先是
sudo apt-get install python-numpy python-scipy python-dev python-pip python-nose g++ libopenblas-dev git
再是
sudo pip install Theano
(这一步骤里,如果你默认是python3,则要改为pip-2.7)
测试下:
Test the newly installed packages
python -c "import numpy; numpy.test()" 约~30s python -c "import scipy; scipy.test()" 约~1m python -c "import theano; theano.test()" 约~30m
Speed test Theano/BLAS
python `python -c "import os, theano; print os.path.dirname(theano.__file__)"`/misc/check_blas.py
测试成功,基本上就能用了。
以下是GPU设置:
检查自己的电脑是否有cuda支持的GPU
lspci | grep -i nvidia
一般得到类似下面的结果
01:00.0 VGA compatible controller: NVIDIA ……
另外,也可用
lspci | grep -i vga
显示所有显卡,用
lspci -v -s 01:00.0
显示特定显卡详细信息。
如果支持,那么安装就行了:
sudo apt-get install nvidia-current sudo apt-get install nvidia-cuda-toolkit
然后主要是设置参数,让优先使用GPU
参见http://deeplearning.net/software/theano/install.html里Using the GPU一节。
主要有两种方法,一种设置THEANO_FLAGS,一种编辑.theanorc文件,以后者为例,
在$home目录(用户主目录,一般是/home/用户名)下创建“ .theanorc”文件,编辑为
[global] device = gpu floatX = float32 [cuda] root = /usr/lib/nvidia-cuda-toolkit //cuda目录一项设置为自己的cuda安装目录即可(内含bin,lib,include子文件夹)
注意,这个文件你创建了,一般会变为不可见,可用"ls -a"命令列出所有文件从而可以看到,可以用gedit编辑。
此外,还要设置cuda的lib子文件夹(64位下可能还要设置lib64子文件夹)位置到环境变量$LD_LIBRARY_PATH,参见http://blog.csdn.net/xsc_c/article/details/23470565(这篇博客cuda的几个入门文章也很不错)。或者这篇http://www.linuxidc.com/Linux/2014-10/107501.htm和http://www.linuxidc.com/Linux/2012-04/58913.htm 这篇。
测试例子在http://deeplearning.net/software/theano/tutorial/using_gpu.html#using-gpu里
from theano import function, config, shared, sandbox import theano.tensor as T import numpy import time vlen = 10 * 30 * 768 # 10 x #cores x # threads per core iters = 1000 rng = numpy.random.RandomState(22) x = shared(numpy.asarray(rng.rand(vlen), config.floatX)) f = function([], T.exp(x)) print f.maker.fgraph.toposort() t0 = time.time() for i in xrange(iters): r = f() t1 = time.time() print ‘Looping %d times took‘ % iters, t1 - t0, ‘seconds‘ print ‘Result is‘, r if numpy.any([isinstance(x.op, T.Elemwise) for x in f.maker.fgraph.toposort()]): print ‘Used the cpu‘ else: print ‘Used the gpu‘
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。