【E-02】內存不足RuntimeError: CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 2.00 GiB total capacity; 1.34 GiB already allocated; 14.76 MiB free; 1.38 GiB reserved in total by PyTorch)

原因一：找到錯誤點，增加以下語句：

with torch.no_grad():

    outputs = Net_(inputs) ---錯誤代碼的位置。

   

原因二：GPU沒有選對

os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2, 3"

查看性能，發現NVIDIA的只有GPU1，所以改成如下：

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

若是不能在GPU上跑，直接改到CPU上跑，修改如下：

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

 

不過，這種問題，還是GPU內存不足引起的。應該購買或租用高性能顯卡，減少時間的浪費。