在没有图形环境的情况下使用matplotlib

在成功地 远程使用 Tensorflow 之后，有一个问题立刻出现了：matplotlib 不能用了。具体来说，plt.plot()、plt.imshow() 之类的函数都会报错。

怎么回事呢？

我们发现，报错信息大致是“TK 没找到”。那就装一个 TK 去。折腾了半天 TK，无果。

等等…… 哪里不对？服务器上没有图形环境啊，用 TK 就是为了弹出那个窗口的。难不成再装个图形去？

解决

Google 一通之后，下面是一个可行的方案：

在 import matpoltlib.pypolt 之前，先 import matplotlib，并更改 matplotlib 所使用的后端：

import matplotlib as mpl

mpl.use('Agg')
import matplotlib.pyplot as plt

这样就可以使用了。

哦你说还是看不到是吧？那就把图片保存出来。比如：

plt.plot(x_points, y_points, 'o', label='Input Data')
plt.savefig("images/plt.png")

然后手动去看图片就好了。