方法1:修改keras的配置文件,~/.keras/keras.json
{
"backend": "tensorflow",
"epsilon": 1e-07,
"floatx": "float32",
"image_data_format": "channels_last"
}
可以直接将backend项修改为tensorflow或theano,如果直接修改有问题,可以在文本中修改再整体拷贝修改该配置文件即可。
方法2:通过命令修改
KERAS_BACKEND = tensorflow python -c "from keras import backend"
这种修改方式只是临时修改了backend,可以在程序中加入
import os
os.environ['KERAS_BACKEND']= 'theano'