方法一:修改keras.json文件
如果Keras 3以前使用过其它的后端,导入过Keras,那么在用户目录下会生成keras.json文件。例如,我前面使用过TensorFlow,现在想切换为yTorch。
当前keras.json文件的内容如下:
将文件中backend": "tensorflow"
修改为"backend": "torch"
:
方法二:在python代码中修改环境变量
在python代码中,导入keras之前,使用os.environ['KERAS_BACKEND'] = 'torch'
将环境变量KERAS_BACKEND的值设置为torch(备注:设置成功以后,只要后端不变,以后的代码就不需要os.environ['KERAS_BACKEND'] = 'torch'
这条语句了):
import os
# 一定要在导入keras之前设置KERAS_BACKEND环境变量
os.environ['KERAS_BACKEND'] = 'torch'
import keras