python - keras 后端设置

Question

我需要将 keras 后端从默认的 tensorflow 更改为 theano。但是我的默认 python 版本是 3.7，它似乎不适用于 keras（导入行崩溃）。所以，我首先要创建一个特定的环境。

按照47263006的建议，使用 anaconda 创建特定的 python 环境后，我执行了以下操作：

vi ~/.keras/keras.json (and change the backend name in it)

但是使用 virtualenv，编辑 keras.json 文件没有效果。因此，我在 python 代码中采用了以下解决方案：

import os
os.environ['KERAS_BACKEND'] = 'theano'

所以我认为后者可能是更通用的解决方案，我尝试将它与我的 anaconda 环境一起使用，但令人惊讶的是 - 这在那里不起作用。

所以我目前的解决方案是，对于 anaconda 编辑 keras.json 文件，对于 virtualenv 使用 os.environ。

是否有更通用的解决方案来设置适用于 conda 和 virtualenv 的 keras 后端？

标签： pythonkerastheano