import tensorflow as tfgpus = tf.config.list_physical_devices("GPU") print(gpus) if gpus:gpu0 = gpus[0] #如果有多個GPU，僅使用第0個GPUtf.config.experimental.set_memory_growth(gpu0, True) #設置GPU顯存用量按需使用tf.config.set_visible_devices([gpu0],"GPU") 《新程序員》：云原生和全面數字化實踐50位技術專家共同創作，文字、視頻、音頻交互閱讀

總結

以上是生活随笔為你收集整理的tensorflow2中如何设置GPU的使用的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。