logo
Loading...

GPU out of memory 問題 - Cupoy

Build SSD_VGG.ipynb 跑 Use_cuda = False 時沒問題,我試著把 B...

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(如圖二),請問是哪裡的設定錯誤 ? 請專家告知應如何設定 ?