如果通過設(shè)置gpu實(shí)現(xiàn)神經(jīng)網(wǎng)絡(luò)訓(xùn)練

發(fā)布時(shí)間：2021-03-03 15:41:17 來源：億速云閱讀：209 作者：Leah 欄目：開發(fā)技術(shù)

這期內(nèi)容當(dāng)中小編將會(huì)給大家?guī)碛嘘P(guān)如果通過設(shè)置gpu實(shí)現(xiàn)神經(jīng)網(wǎng)絡(luò)訓(xùn)練，文章內(nèi)容豐富且以專業(yè)的角度為大家分析和敘述，閱讀完這篇文章希望大家可以有所收獲。

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

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

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

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

如果通過設(shè)置gpu實(shí)現(xiàn)神經(jīng)網(wǎng)絡(luò)訓(xùn)練

上述就是小編為大家分享的如果通過設(shè)置gpu實(shí)現(xiàn)神經(jīng)網(wǎng)絡(luò)訓(xùn)練了，如果剛好有類似的疑惑，不妨參照上述分析進(jìn)行理解。如果想知道更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道。

向AI問一下細(xì)節(jié)

如果通過設(shè)置gpu實(shí)現(xiàn)神經(jīng)網(wǎng)絡(luò)訓(xùn)練

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽