tensorflow使用GPU訓練時的顯存佔用問題

http://blog.csdn.net/sunwuhuang1/article/details/53946462 這篇文章很有用。 最近用GoogleNet (inception-v3)做fine-tune訓練時,遇到了一個問題。當我選擇單個GPU訓練時候,服務器中所有的GPU顯存都會被佔滿,如下圖所示:     出現以上問題的原因是tensorflow在訓練時默認佔用所有GPU的顯存。    
相關文章
相關標籤/搜索