原文:OOM when allocating tensor of shape [] and type float [[node conv2d_224/kernel/Initializer/random_uniform/min (defined at ./intances/utils.py:19) ]]

當你們在用模型加載,tensorflow gpu預測的時候,有沒有出現這樣的錯誤 經過網上的資料查閱,更多的解釋是GPU的顯存不足造成的,於是乎,我查看了一下GPU顯存:nvidia smi 不看不知道,一看確實是那么回事:tensorflow gpu執行默認將GPU顯存占滿,官網也做了解釋說明:https: www.tensorflow.org guide using gpu 剛開始沒有看到官 ...

2019-05-29 11:34 0 757 推薦指數:

查看詳情

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