keras修改backend的简单方法

编辑: admin 分类: python 发布时间: 2021-12-24 来源:互联网

方法1

在users文件夹下新建.keras文件夹，在文件夹下新建keras.json文件，json内容为：

{
    "image_dim_ordering": "tf", 
    "epsilon": 1e-07, 
    "floatx": "float32", 
    "backend": "tensorflow"
}

这其中的backend可以是tensorflow或者theano,当然前提是你已经安装好了。

但是上面这种方式有一个弊端就是每次当你想切换backend的时候都需要手动修改这个文件，所以有一种能够动态修改的方式不是更好吗？请看下面的方法：

方法2

import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
import keras.backend as K
K.set_image_dim_ordering('tf')

首先载入os库，将keras的backend修改为tensorflow，再将keras的backend加载，设置order为tensorflow格式。

补充：解决引入keras后出现的Using TensorFlow backend的错误

问题：

用Anaconda检索keras-gpu，apply后，理论上应该自动配好相关的cuddn和tensorflow-gpu及其相应版本，但运行程序时出现如标题错误和一堆报错。

按网上一堆方法改，都不好用。比如，加os.environ['KERAS_BACKEND']='tensorflow-gpu'，更新setup tool等。

解决：

最后误打误撞，在Anaconda上修改tensorflow-gpu版本由之前默认的1.10改成1.80

python3.6

以上为个人经验，希望能给大家一个参考，也希望大家多多支持hwidc。

【出处：http://www.1234xp.com/tw.html 转载请说明出处】
更多阅读