如果从命令行启动脚本,我使用“KERAS_BACKEND”环境变量在 Theano 和 Tensoflow 之间切换。如果脚本在笔记本中运行,可以做些什么来切换后端?
Keras:在笔记本中切换后端
数据挖掘
机器学习
2022-03-02 23:47:39
1个回答
您可以通过 python 的内置os模块设置环境变量。
import os
os.environ["KERAS_BACKEND"] = "tf" # or theano
在一个笔记本中进行更改似乎不会延续到在同一 Jupyter 会话中启动的其他笔记本(从单个终端运行)。
在这里我尝试获取环境变量notebook 2,我们看到它不存在:

在笔记本 1 中创建环境变量:
现在进行更改notebook 1(在确认环境变量不存在之后 - 我们捕获异常)
现在回去notebook 2看看它是否接收到改变的值:
Keras 在导入时还应读取一些设置。看看文档。如果您在 Windows 上,您应该在那里查看一些单独的步骤。
其它你可能感兴趣的问题

