Runtime Error:cuda runtime error(2):out of memory at /……/torch/…….cu:66

在运行faster RCNN_pytorch的代码时碰到了这个问题,找到以下几种解决方法:

1.减小batchsize
如果是因为模型太大内存不够,减小bs就能解决
2.Variable(x)改成Variable(x,volatile=True)
3.在模型中预测部分的代码前面加入with torch.no_grad():

发布了10 篇原创文章 · 获赞 0 · 访问量 65

猜你喜欢

转载自blog.csdn.net/qq_42499257/article/details/105678410