设置Keras的后端

时间:2019-02-06 05:08:37

标签: python keras theano

我需要将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 env一起使用,但是很奇怪-在那里没有用。

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

是否存在一种更通用的设置keras后端的解决方案,该后端对conda和virtualenv都适用?

0 个答案:

没有答案