解决方法

添加代码，指定不需要GUI的backend

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