RuntimeError('Invalid DISPLAY variable')

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

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

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