TensorFlow使用GPU訓練網絡時多塊顯卡的顯存使用問題

剛開始使用實驗室的服務器網絡訓練時，有這樣的現象： 能看到，只使用單個GPU跑程序，但三塊顯卡的顯存都被佔用。html 這是由於TensorFlow訓練時默認佔用全部GPU的顯存。引自TF官網：web By default, TensorFlow maps nearly all of the GPU memory of all GPUs (subject to CUDA_VISIBLE_DEVIC

>>阅读原文<<