下述程式碼源於參考TensorFlow入門一中線性迴歸的實現，想將結果圖片儲存檢視。

1、匯入檔案：

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

File "/usr/lib64/python2.7/lib-tk/Tkinter.py", line 1745, in __init__
    self.tk = _tkinter.create(screenName, baseName, className, interactive, wantobjects, useTk, sync, use)
_tkinter.TclError: no display name and no $DISPLAY environment variable
 

2、準備繪圖：

因為第一次使用python繪圖，我先是寫下以下語句想畫個500*500的視窗：

plt.figure(figsize=(500, 500)) 

plt.plot(train_X, train_Y, 'ro', label='Original data')
plt.plot(train_X, sess.run(W) * train_X + sess.run(b), label='Fitted line')
plt.savefig("examples.jpg")

plt.figure(figsize=(500, 500)) 
 

but，我還不知道why，希望後續可以理解。。。