出現這個的原因是gpu顯存不夠導致的,一般是我們設置了程序根據需求增長導致的
這里,我們設置 allow_growth=False 就可以控制顯存使用的增長,從而控制顯存的使用,而不會程序運行一半報錯。
這種情況是建立在我們的顯存比較小的情況下使用的策略,如果說我們顯存夠大則不用擔心這種問題。
出現這個的原因是gpu顯存不夠導致的,一般是我們設置了程序根據需求增長導致的
這里,我們設置 allow_growth=False 就可以控制顯存使用的增長,從而控制顯存的使用,而不會程序運行一半報錯。
這種情況是建立在我們的顯存比較小的情況下使用的策略,如果說我們顯存夠大則不用擔心這種問題。
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。