TensorFlow 2.0 經常使用模塊7:tf.config

在上一篇文章中,咱們介紹了 tf.TensorArray 的創建方式。html 原本接下來將介紹 TensorFlow 中的經常使用模塊 tf.config,如何合理分配顯卡資源,設置顯存使用策略。web 指定當前程序使用的 GPU微信 不少時候的場景是:實驗室 / 公司研究組裏有許多學生 / 研究員須要共同使用一臺多 GPU 的工做站,而默認狀況下 TensorFlow 會使用其所可以使用的全部
相關文章
相關標籤/搜索