RuntimeError('Invalid DISPLAY variable')

在服务器上跑的时候遇到了这个问题,原因在于sever不能显示,必须要保存下来,但是如果为了,程序能够跑起来,可以添加以下代码:

RuntimeError('Invalid DISPLAY variable')

plt.switch_backend('agg') after import matplotlib.pyplot as plt

参考:
https://github.com/matplotlib/matplotlib/issues/3466/