tensorflow顯存管理、tensorflow使用多個gpu訓練

 一般在程序開始以前並不知道須要多大的顯存,程序會去申請GPU的顯存的50%html 好比一個8G的內存,被佔用了2G,那麼程序會申請4G的顯存(由於有足夠的剩餘顯存)python 若是此時顯存被佔用7G,那麼程序會申請剩下的全部的1G的顯存。也許你的程序根本用不着這麼多顯存,200M也許就夠了,這時候若是程序能按照需求去申請就行了,幸運的是,這樣的方法是存在的:bash import tenso
相關文章
相關標籤/搜索