码迷,mamicode.com
首页 > 系统相关 > 详细

ubuntu14.04安装theano

时间:2015-04-13 18:04:48      阅读:440      评论:0      收藏:0      [点我收藏+]

标签:

基本上按照官网来就行:

先是

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.htmhttp://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‘

  

ubuntu14.04安装theano

标签:

原文地址:http://www.cnblogs.com/rolling-stone/p/4422515.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!