使用matplotlib时Linux批处理任务执行失败的问题排查

为什么手动执行正常，batch任务里DISPLAY无效？ #

当你手动登录终端时，处于交互式图形会话中，系统会自动为你设置$DISPLAY环境变量（通常是:0，对应当前显示器），而matplotlib默认的TkAgg后端依赖这个变量来创建图形上下文。

但batch命令是在后台非交互式环境执行任务的，它不会关联任何物理或虚拟显示器，所以环境变量里根本没有$DISPLAY，这就导致matplotlib找不到图形输出目标，直接抛出_tkinter.TclError错误。

排查步骤 #

1. 对比环境变量差异
   在batch_job.txt里添加一行命令，导出batch环境的所有变量：

   env > batch_env.txt
   python plot.py 2>&1 | tee output.txt
   

   然后手动在终端执行env > interactive_env.txt，对比两个文件，你会发现batch_env.txt里完全没有DISPLAY项，而interactive_env.txt里存在该变量。

2. 确认matplotlib后端
   手动执行以下命令，查看当前matplotlib使用的后端：

   python -c "import matplotlib; print(matplotlib.get_backend())"
   

   大概率会输出TkAgg——这就是需要图形环境的后端。

解决方法 #

根据你的场景（仅需保存图片，无需交互显示），推荐两种简单解决方案：

方案1：在代码中指定无图形后端

直接在plot.py开头添加两行代码，强制matplotlib使用不需要图形环境的Agg后端：

import matplotlib
matplotlib.use('Agg')  # 必须放在import plt之前
import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0,4)
y = np.sin(x)
plt.plot(x,y)
plt.savefig("test.png")
print "done"

方案2：通过环境变量指定后端（无需修改代码）

修改batch_job.txt，添加MPLBACKEND=Agg环境变量，让matplotlib执行时自动切换后端：

MPLBACKEND=Agg python plot.py 2>&1 | tee output.txt

方案3：虚拟显示器（适合需要图形交互的场景）

如果后续有图形交互需求，可以安装Xvfb虚拟显示器：

apt-get install xvfb

然后修改batch_job.txt：

Xvfb :1 -screen 0 1024x768x16 &
export DISPLAY=:1
python plot.py 2>&1 | tee output.txt

不过这个方案对你当前场景冗余，前两个方案足以解决问题。

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