赞
踩
在服务器上运行程序的时候出现了这个错误,先用nvidia-smi看一下目前现存使用情况。
可以看到显存还有。
解决办法:把batch size改小一点。虽然显存还有但是不一定够用。原来使用的是32,改成4后可以运行了。