在定義圖結構之前不用加入gpu：0，只有在session中計算之前在加入，否則的話會提示不能使用gpu保存模型等問題。

 with
  tf.device(
 '/gpu:0'
  ):

補充：關于應用gpu訓練神經網絡的注意事項

對于GPU來說，一定要注意的是，要分別在兩個GPU上，或者不同時的在一個GPU上運行train和evaluation的部分，否則限于GPU擅長迭代而不擅長邏輯的特性，會發生OOM（out of memory）

以上為個人經驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。