1. OSError: [Error 12] Cannot allocate memory

解決辦法： 出現這個錯誤時我是絕望的。。。因為我看了別人的解決辦法，要加內存條才能解決。。。但是我不甘心，想了很久——發現了一個解決辦法：將 DataLoader 中的 num_workers設置為1。缺點：數據加載必然變得很慢，可憐我的內存條只有 8 個G

2. RuntimeError: CUDA error: device-side assert triggered

解決辦法： 檢查類別數目設置是否有誤！標簽分類應該從 0 開始，如果共有10類，則類別應是從 0 開始直到 19

3. CUDA error: out of memory

解決辦法： 可以適當縮小 batch size 來解決

轉載于:https://www.cnblogs.com/xxxxxxxxx/p/11494734.html

總結

以上是生活随笔為你收集整理的利用GPU训练网络时遇到的一些问题的全部內容，希望文章能夠幫你解決所遇到的問題。

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