【问题标题】:Why is theano running so slow?为什么theano运行这么慢?
【发布时间】:2016-06-13 11:02:59
【问题描述】:

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

【问题讨论】:

  • 你能分解一下,看看哪个步骤花费了时间吗?可能是 f = function(...),编译发生的地方,但最好确定一下。
  • 我运行了同一段代码,它运行不到一秒钟。您是否更新了机器上的 NumPy 和 Theano 版本?
  • f = function([x, y], z) 占用了超过 95% 的时间,其他一切运行顺利

标签: theano


【解决方案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 用户的生活更轻松。 "

确切的标志是这样的:

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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2019-12-11
    • 2014-08-31
    • 2011-10-18
    • 1970-01-01
    • 2021-09-03
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多