Cuda out of memory显存不足

在服务器上运行程序的时候出现了这个错误，先用nvidia-smi看一下目前现存使用情况。

可以看到显存还有。

解决办法：把batch size改小一点。虽然显存还有但是不一定够用。原来使用的是32，改成4后可以运行了。