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

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

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

對(duì)于GPU來(lái)說(shuō)，一定要注意的是，要分別在兩個(gè)GPU上，或者不同時(shí)的在一個(gè)GPU上運(yùn)行train和evaluation的部分，否則限于GPU擅長(zhǎng)迭代而不擅長(zhǎng)邏輯的特性，會(huì)發(fā)生OOM（out of memory）

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