直接在脚本中配置 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"
我是 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"