最近使用tensorflow構建了一個淺層的網絡，初始化參數的時候發現兩個1080ti 22G的顯存馬上占滿
后來發現解決辦法為：在創建session的時候把顯存設置為自適應即可，如下
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(graph=graph, config=config)
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。