直接在脚本中配置 Theano

Theano config directly in script

我是 Theano 的新手,我想知道如何直接从脚本配置默认设置(不设置 envir.variables)。例如。这是一个可行的解决方案 (source):

$ THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python check1.py

我打算提出仅由以下人员执行的相同解决方案:

$ python check1.py

并且其他参数直接在脚本本身中设置。例如。有点像这样:

import theano
theano.set('mode', 'FAST_RUN')
theano.set('device', 'gpu')
theano.set('floatX', 'float32')
# rest of the script

这可能吗?我阅读了config page,它提供了允许我读取已经设置的值的信息(但不是我自己设置的)。

当你这样做时:

$ THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python check1.py

您实际上所做的只是在 运行 Python 脚本之前设置一个环境变量。

您也可以在 Python 中设置环境变量。例如,THEANO_FLAGS 环境变量可以在 Python 内设置,如下所示:

import os
os.environ["THEANO_FLAGS"] = "mode=FAST_RUN,device=gpu,floatX=float32"

请注意,导入 Theano 后无法更改某些 Theano 配置变量,所以这很好:

import os
os.environ["THEANO_FLAGS"] = "mode=FAST_RUN,device=gpu,floatX=float32"
import theano

但这不会按预期工作:

import theano
import os
os.environ["THEANO_FLAGS"] = "mode=FAST_RUN,device=gpu,floatX=float32"