Keras:在笔记本中切换后端

数据挖掘 机器学习
2022-03-02 23:47:39

如果从命令行启动脚本,我使用“KERAS_BACKEND”环境变量在 Theano 和 Tensoflow 之间切换。如果脚本在笔记本中运行,可以做些什么来切换后端?

1个回答

您可以通过 python 的内置os模块设置环境变量。

import os

os.environ["KERAS_BACKEND"] = "tf"   # or theano

在一个笔记本中进行更改似乎不会延续到在同一 Jupyter 会话中启动的其他笔记本(从单个终端运行)。

在这里我尝试获取环境变量notebook 2,我们看到它不存在: 笔记本 2

在笔记本 1 中创建环境变量:

现在进行更改notebook 1(在确认环境变量不存在之后 - 我们捕获异常)

笔记本 1

现在回去notebook 2看看它是否接收到改变的值:

再次笔记本 2


Keras 在导入时还应读取一些设置。看看文档如果您在 Windows 上,您应该在那里查看一些单独的步骤。