Ép matplotlib không dùng DISPLAY mà lưu hình ra file (hữu ích khi chạy trên server không có giao diện đồ họa):

import matplotlib
matplotlib.use('Agg')  # phải gọi trước khi import pyplot
import matplotlib.pyplot as plt
# ...
plt.savefig("plt.png")
plt.close()