cuda runtime error (2) : out of memory解决

最新推荐文章于 2024-08-17 22:11:19 发布

H_航

最新推荐文章于 2024-08-17 22:11:19 发布

阅读量6.9k

点赞数 2

分类专栏：深度学习图像处理

本文链接：https://blog.csdn.net/Harid/article/details/108169768

版权

深度学习图像处理专栏收录该内容

1 篇文章 0 订阅

订阅专栏

cuda runtime error (2) : out of memory解决

在这里插入图片描述
报错如图
一般遇到out of memory的情况都是服务器的容器上在跑别的程序或是内存没释放，模型太大等，不过这次出现这个错误用nvidia-smi试了下，容器只用了10MB，模型也是不可能超内存的。后来查了下应该在DataLoader中加入pin_memory=False，改完之后，程序立马跑通。