tensorflow初始化參數內存佔滿問題

最近使用tensorflow構建了一個淺層的網絡，初始化參數的時候發現兩個1080ti 22G的顯存立刻佔滿

後來發現解決辦法爲：在建立session的時候把顯存設置爲自適應便可，以下

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

session = tf.Session(graph=graph, config=config)