theano - 为什么 theano 运行这么慢？

Question

我是 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 内存，同时运行的进程并不多。

Answer 1

首先检查 Theano 标志。

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

——

但是，请在此处阅读:

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

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

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

由于您必须从 IDE 运行，您必须编辑 .theanorc

如上面 Theano 链接所述:

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

确切的标志是这样的:

配置模式

字符串值:“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