在Windows上将Keras后端从Theano更改为TensorFlow,可以按照以下步骤操作:
keras.json
文件keras.json
文件:.keras
文件夹中,例如:C:\Users\YourUsername\.keras\keras.json
。python -c "import keras; print(keras.__file__)"
来定位Keras的安装位置。keras.json
文件:keras.json
文件,你会看到类似以下的内容: { "floatx": "float32", "epsilon": 1e-07, "backend": "theano", "image_data_format": "channels_last" }"backend"
的值从"theano"
改为"tensorflow"
: { "floatx": "float32", "epsilon": 1e-07, "backend": "tensorflow", "image_data_format": "channels_last" }keras.json
文件的修改,并重启你的Python环境(例如Jupyter Notebook内核或Python解释器)。KERAS_BACKEND
tensorflow
如果你不想修改配置文件或设置环境变量,也可以在Python代码中动态地设置后端:
import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
import keras
将上述代码放在你的脚本或Jupyter Notebook的最开始部分即可。
领取专属 10元无门槛券
手把手带您无忧上云