GPU out of memory 問題
2020/01/06 05:33 下午
電腦視覺深度學習討論版
胡連福
觀看數：0
回答數：1
收藏數：0
Build SSD_VGG.ipynb 跑 Use_cuda = False 時沒問題，
我試著把 Build SSD_VGG.ipynb 的 Use_cuda = True 設為 True，使用 GPU 來 run，但遇到以下out of memory問題 :
RuntimeError: CUDA out of memory. Tried to allocate 88.00 MiB (GPU 0; 2.00 GiB total capacity; 1.26 GiB already allocated; 17.00 MiB free; 19.23 MiB cached)
====================================
1. 我使用的 NB 內是用 Nvida-MX150 GPU，我用 nvidia-smi 查詢 memory 只顯示 2048 MiB(如圖一)，但我的系統 DRAM 已擴充到 16 GB(如圖二)，請問是哪裡的設定錯誤 ? 請專家告知應如何設定 ?