1. OSError: [Error 12] Cannot allocate memory

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

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

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

3. CUDA error: out of memory

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

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

總結

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。