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

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

補(bǔ)充：關(guān)于應(yīng)用gpu訓(xùn)練神經(jīng)網(wǎng)絡(luò)的注意事項

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

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