您好,登錄后才能下訂單哦!
這期內容當中小編將會給大家帶來有關如果通過設置gpu實現神經網絡訓練,文章內容豐富且以專業的角度為大家分析和敘述,閱讀完這篇文章希望大家可以有所收獲。
在定義圖結構之前不用加入gpu:0,只有在session中計算之前在加入,否則的話會提示不能使用gpu保存模型等問題。
with tf.device( '/gpu:0' ):
補充:關于應用gpu訓練神經網絡的注意事項
對于GPU來說,一定要注意的是,要分別在兩個GPU上,或者不同時的在一個GPU上運行train和evaluation的部分,否則限于GPU擅長迭代而不擅長邏輯的特性,會發生OOM(out of memory)
上述就是小編為大家分享的如果通過設置gpu實現神經網絡訓練了,如果剛好有類似的疑惑,不妨參照上述分析進行理解。如果想知道更多相關知識,歡迎關注億速云行業資訊頻道。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。