RuntimeError: CUDA out of memory. Tried to allocate 14763.13 GiB (GPU 3; 10.73 GiB total capacity; 165.28 MiB already allocated; 9.61 GiB free; 10.72 MiB cached)

原因：要將大量數據加載到GPU上時出現的錯誤，數據量太大了
解決辦法：做數據的切分，一個batch一個batch的分批計算。在pytorch中有個Dataset類，可以自己定義一個加載數據的類，繼承Dataset，同時實現Dataset的兩個方法：getitem() 和 len() ，然后使用 Dataloader 對自定義的類進行包裝后，在訓練階段就可以讓pytorch自動為你批量加載數據了，具體的實現細節參考我的另一篇文章 Pytorch數據讀取(Dataset, DataLoader, DataLoaderIter)

總結

以上是生活随笔為你收集整理的CUDA out of memory. Tried to allocate 14763.13 GiB (GPU 3; 10.73 GiB total capacity; 165.28 MiB alre的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。