代码之家  ›  专栏  ›  技术社区  ›  R71

设置keras的后端

  •  1
  • R71  · 技术社区  · 6 年前

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

    使用anaconda创建特定的python环境之后 47263006 ,我做了以下操作:

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

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

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

    所以我想也许后者是更普通的来源,我试着用它来对付我的水蟒环境,但令人惊讶的是-这在那里不起作用。

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

    有没有一个更通用的解决方案来设置keras后端,它将同时适用于conda和virtualenv?

    0 回复  |  直到 6 年前