在服務器上運行程序的時候出現了這個錯誤，先用nvidia-smi看一下目前現存使用情況。

可以看到顯存還有。

解決辦法：把batch size改小一點。雖然顯存還有但是不一定夠用。原來使用的是32，改成4后可以運行了。

總結

以上是生活随笔為你收集整理的Cuda out of memory显存不足的全部內容，希望文章能夠幫你解決所遇到的問題。

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