tensorflow ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[256,256,15,15] and type float on 報錯(顯存不夠報錯)


出現這個的原因是gpu顯存不夠導致的,一般是我們設置了程序根據需求增長導致的

 

 

這里,我們設置 allow_growth=False 就可以控制顯存使用的增長,從而控制顯存的使用,而不會程序運行一半報錯。

這種情況是建立在我們的顯存比較小的情況下使用的策略,如果說我們顯存夠大則不用擔心這種問題。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM