如何在通过Supervisor加载时动态设置Python中的Keras后端？

我之前也踩过这个一模一样的坑！问题根源在于Keras初始化时会优先读取系统/进程的环境变量，然后才会处理代码里的设置——如果Supervisor的运行环境里已经预设了KERAS_BACKEND=tensorflow，那你的代码里再改就已经晚了，根本不会生效。

下面是几个亲测有效的解决办法，你可以根据自己的场景选：

方法1：修改Supervisor配置，移除/调整环境变量 #

找到你对应Python脚本的Supervisor配置文件（通常在/etc/supervisor/conf.d/目录下），定位到environment配置项，把其中的KERAS_BACKEND=tensorflow删掉，或者改成你需要的后端。示例配置：

[program:your_python_script]
command=python /opt/your_script.py
directory=/opt/script_dir
user=ubuntu
# 移除或修改包含KERAS_BACKEND的环境变量设置
environment=PATH="/usr/local/bin",PYTHONPATH="/opt/venv/lib/python3.8/site-packages"

修改后记得重启Supervisor服务使配置生效：

sudo supervisorctl reread
sudo supervisorctl update
sudo supervisorctl restart your_python_script

方法2：在Python脚本最开头强制设置环境变量 #

如果没法修改Supervisor的全局配置，那就在你的Python脚本最最开头（一定要在导入Keras之前）设置环境变量，这样能覆盖进程级的环境变量。示例代码：

# 必须放在导入Keras的代码之前！
import os
os.environ['KERAS_BACKEND'] = 'theano'  # 替换成你需要的后端

# 之后再导入Keras
import keras
from keras import backend as K

# 可以加一行验证是否生效
print(f"当前Keras后端：{K.backend()}")

方法3：在Supervisor启动命令中临时覆盖环境变量 #

还有一种灵活的方式，直接在Supervisor的command里通过bash命令临时设置环境变量，这样只会影响当前脚本的运行环境：

[program:your_python_script]
# 用bash -c来包裹命令，临时设置KERAS_BACKEND
command=bash -c "KERAS_BACKEND=theano python /opt/your_script.py"
directory=/opt/script_dir
user=ubuntu

不管用哪种方法，都建议在脚本里加一行打印后端的代码，确认设置是否生效——毕竟坑踩多了，验证一下才放心😉

内容的提问来源于stack exchange，提问作者MenorcanOrange