因为我是在服务器的虚拟环境下跑代码,所以修改home/user/下的keras.json文件并没有效果。
看网上大佬说是修改json文件可以切换keras后端的。
{ # keras.json文件内容
"image_dim_ordering": "tf",
"epsilon": 1e-07,
"floatx": "float32",
"backend": "theano"
}
就是修改backend这里的值。
以上方法都试过,没有作用,记录一下以防以后碰到同样的问题。
最后通过在python代码中import keras前加入一个环境变量修改的语句
import os
os.environ['KERAS_BACKEND']='tensorflow'
完美解决了。
网友评论