为什么theano跑得这么慢?

时间:2016-06-11 21:39:28

标签: theano

我是Theano的新手并尝试了一些例子。

import numpy
import theano.tensor as T
from theano import function
import datetime
print datetime.datetime.now()
x = T.dscalar('x')
y = T.dscalar('y')
z = x + y
f = function([x, y], z)
print f(2, 3)
print numpy.allclose(f(16.3, 12.1), 28.4)
print datetime.datetime.now()

运行它需要15分钟。我使用的是2GB内存,并没有很多进程同时运行。

1 个答案:

答案 0 :(得分:4)

首先检查Theano Flags。

如果你没有运行: THEANO_FLAGS = mode = FAST_RUN 或者没有使用默认标志运行,或者你更改了.theanorc,可能需要一些时间。

-

但请阅读:

http://deeplearning.net/software/theano/tutorial/using_gpu.html

您还可以在此处查看有关Theano Flags的更多信息:

http://deeplearning.net/software/theano/library/config.html

由于您必须从IDE运行,因此您必须编辑 .theanorc

如上面的Theano Link所述:

” 默认为$ HOME /.theanorc。在Windows上,它默认为$ HOME / .theanorc:$ HOME / .theanorc.txt,以使Windows用户的生活更轻松。 “

确切的标志是这样的:

<强> config.mode

字符串值:'Mode','ProfileMode'(不建议使用),'DebugMode','FAST_RUN','FAST_COMPILE'

如果这没有帮助,请务必 将Theano更新到最前沿 并使用theanorc设置 编辑您的问题 < /强>!

pip install --upgrade --no-deps git + git://github.com/Theano/Theano.git

您还可以使用OpenMP来使用额外的线程/核心:

THEANO_FLAGS = mode = FAST_RUN THEANO_FLAGS ='openmp = True'OMP_NUM_THREADS = 4 python x.py