问题描述:在本地的pycharm可以正常使用matplotlib绘图,但在ssh远程绘图时会报错:RuntimeError: Invalid DISPLAY variable
错误原因:
matplotlib的默认backend是TkAgg,而
FltkAgg, GTK, GTKAgg, GTKCairo, TkAgg , Wx or WxAgg
这几个backend都要求有GUI图形界面的,所以在ssh操作的时候会报错
解决方法:
在import matplotlib时,指定所用的backend为不需要用GUI的backend,不需要用GUI的backend有:
Agg、Cairo、PS、PDF和SVG
import matplotlib.pyplot as plt
plt.switch_backend('agg')