相信大部分使用pytorch跑程序的朋友都在服务器上遇到过这个问题：runoutofmemory，其实就是内存不够用enough1.bug提示的时候专门提示某个GPU使用了多少内存。如果剩余内存不够用，只需要减小batch_size2即可。不管batch_size多小，还是会报错：runoutofmemory。这是因为你的pytorch版本太高了。这时在withtorch.no_grad()中加入如下代码：output=net(input,inputcoord)3.（大坑）如果没有提示内存已经用了多少，内存还剩多少，那就是可能是因为你的pytorch版本和cuda版本不匹配，那么你可以在终端命令行输入如下代码：importtorchprint(torch.__version__)print(torch.version.cuda)print(torch.backends.cudnn.version())print(torch.cuda.is_available())通过上面的代码可以查看你当前的torch版本，cuda版本，cudnn版本，以及torch是否可以在当前cuda版本下使用gpu。如果返回false，建议调整cuda版本或者pytorch版本；。...
